某编程大赛中设计有一个挑战环节，选手可以查看其他选手的代码，发现错误后，提交一组测试数据将对手挑落马下。为了减小被挑战的几率，有些选手会故意将代码写得很难看懂，比如把所有回车去掉，提交所有内容都在一行的程序，令挑战者望而生畏。
为了对付这种选手，现请你编写一个代码排版程序，将写成一行的程序重新排版。当然要写一个完美的排版程序可太难了，这里只简单地要求处理C语言里的for、while、if-else这三种特殊结构，而将其他所有句子都当成顺序执行的语句处理。输出的要求如下：
默认程序起始没有缩进；每一级缩进是 2 个空格；
每行开头除了规定的缩进空格外，不输出多余的空格；
顺序执行的程序体是以分号“;”结尾的，遇到分号就换行；
在一对大括号“{”和“}”中的程序体输出时，两端的大括号单独占一行，内部程序体每行加一级缩进，即：
{
程序体
}
for的格式为：
for (条件) {
程序体
}
while的格式为：
while (条件) {
程序体
}
if-else的格式为：
if (条件) {
程序体
}
else {
程序体
}
输入格式：
输入在一行中给出不超过 331 个字符的非空字符串，以回车结束。题目保证输入的是一个语法正确、可以正常编译运行的 main 函数模块。
输出格式：
按题面要求的格式，输出排版后的程序。
输入样例：
int main() {int n, i; scanf("%d", &n);if( n>0)n++;else if (n<0) n--; else while(n<10)n++; for(i=0; i<n; i++ ){ printf("n=%d\n", n);}return 0; }
输出样例：
int main()
{
int n, i;
scanf("%d", &n);
if ( n>0) {
n++;
}
else {
if (n<0) {
n--;
}
else {
while (n<10) {
n++;
}
}
}
for (i=0; i<n; i++ ) {
printf("n=%d\n", n);
}
return 0;
}
格式要求挺高的，代码写的很乱，先把输入的字符串中if,for,while,else的格式进行调整该加大括号加大括号，空格数量需要修剪，都是一个空格，然后按照要求输出，对于小括号里的直接输出，每行开头结尾没有多余空格。左大括号左边不是判断语句需要换行。
代码:
#include <iostream>
#include <cstdio>
using namespace std;
int ignore_space(const string &s,int d,int t) {
while(d < s.size() && d > && s[d] == ' ') {
d += t;
}
return d;
}
int trim_space(string &s,int d,int t) {
if(s[d] != ' ') s.insert(d," ");
d += t;
int i = ;
while(i + d < s.size() && s[i + d] == ' ') i ++;
if(i) s.erase(d,i);
return d;
}
bool judge(string s,int d,int &flag) {
if(d && isalpha(s[d - ])) return false;
if(s.substr(d,) == "if" && (s[d + ] == ' ' || s[d + ] == '(')) return true;
if(s.substr(d,) == "for" && (s[d + ] == ' ' || s[d + ] == '(')) return true;
if(s.substr(d,) == "while" && (s[d + ] == ' ' || s[d + ] == '(')) return true;
flag = ;
if(s.substr(d,) == "else" && (s[d + ] == ' ' || s[d + ] == '{')) return true;
return false;
}
void perfect(string &s) {
for(int i = s.size() - ;i >= ;i --) {
int ise = ;
if(s[i] == ')') {
int d = ,j = i - ;
while(d && j >= ) {
if(s[j] == ')') d ++;
else if(s[j] == '(') d --;
j --;
}
i = j + ;
}
else if(judge(s,i,ise)) {
int j = i;
while(j < s.size() && isalpha(s[j])) j ++;
if(ise) {
j = trim_space(s,j,);
int d = ;
while(d != && ++ j < s.size()) {
if(s[j] == '(') d ++;
else if(s[j] == ')') d --;
}
j ++;
}
j = trim_space(s,j,);
if(s[j] != '{') {
s.insert(j,"{");
j ++;
int flag = ;
while(j < s.size()) {
if(s[j] == ';' && flag == ) {
s.insert(j + ,"}");
break;
}
else if(s[j] == '(') flag ++;
else if(s[j] == ')') flag --;
else if(s.substr(j,) == "if") flag ++;
else if(s.substr(j,) == "else") flag --;
else if(s[j] == '{') flag ++;
else if(s[j] == '}') {
flag --;
if(flag == && s[ignore_space(s,j + ,)] != 'e') flag --;
if(flag == ) {
s.insert(j + ,"}");
break;
}
}
j ++;
}
}
}
}
}
void put_space(int k) {
while(k --) {
printf(" ");
}
}
bool check(char s) {
return s == ')' || s == 'e' || s == '{' || s == '}' || s == ';';
}
int main() {
int flag = ,space_num = ,start = ;
string s;
getline(cin,s);
perfect(s);
for(int i = ignore_space(s,,);i < s.size();i ++) {
if(start) {
start = ;
if(s[i] == '}') put_space(space_num - );
else put_space(space_num);
}
if(s[i] == '{') {
int d = ignore_space(s,i - ,-);
if(!flag || !check(s[d])) {
putchar('\n');
put_space(space_num);
}
flag = ;
putchar(s[i]);
putchar('\n');
space_num ++;
start = ;
i = ignore_space(s,i + ,) - ;
}
else if(s[i] == '}') {
space_num --;
putchar(s[i]);
i = ignore_space(s,i + ,) - ;
if(s[i + ] == '}') {
putchar('\n');
start = ;
}
else if(i < s.size() - ) {
putchar('\n');
start = ;
}
}
else if(s[i] == '(') {
int d = ,j = i;
putchar(s[i]);
while(d) {
putchar(s[++ j]);
if(s[j] == '(') d ++;
else if(s[j] == ')') d --;
}
i = j;
if(!flag) i = ignore_space(s,i + ,) - ;
}
else if(s[i] == ';') {
putchar(s[i]);
i = ignore_space(s,i + ,) - ;
putchar('\n');
start = ;
}
else {putchar(s[i]);}
}
return ;
}