某编程大赛中设计有一个挑战环节，选手可以查看其他选手的代码，发现错误后，提交一组测试数据将对手挑落马下。为了减小被挑战的几率，有些选手会故意将代码写得很难看懂，比如把所有回车去掉，提交所有内容都在一行的程序，令挑战者望而生畏。
为了对付这种选手，现请你编写一个代码排版程序，将写成一行的程序重新排版。当然要写一个完美的排版程序可太难了，这里只简单地要求处理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) {
while(d < s.size() && s[d] == ' ') {
d ++;
}
return d;
}
int trim_space(string &s,int d) {
if(s[d] != ' ') s.insert(d," ");
d ++;
int i = 0;
while(i + d < s.size() && s[i + d] == ' ') i ++;
if(i) s.erase(d,i);
return d;
}
bool judge(string s,int d) {
if(s.substr(d,2) == "if" && (s[d + 2] == ' ' || s[d + 2] == '(')) return true;
if(s.substr(d,3) == "for" && (s[d + 3] == ' ' || s[d + 3] == '(')) return true;
if(s.substr(d,5) == "while" && (s[d + 5] == ' ' || s[d + 5] == '(')) return true;
return false;
}
void perfect(string &s) {
for(int i = s.size() - 1;i >= 0;i --) {
if(s[i] == ')') {
int d = 1,j = i - 1;
while(d && j >= 0) {
if(s[j] == ')') d ++;
else if(s[j] == '(') d --;
j --;
}
i = j + 1;
}
else if(judge(s,i)) {
for(int j = i;j < s.size();j ++) {
if(!isalpha(s[j])) {
j = trim_space(s,j);
int d = 1;
j ++;
while(d) {
if(s[j] == '(') d ++;
else if(s[j] == ')') d --;
j ++;
}
j = trim_space(s,j);
if(s[j] != '{') {
s.insert(j,"{");
j ++;
int flag = 0;
while(j < s.size()) {
if(s[j] == ';' && flag == 0) {
s.insert(j + 1,"}");
break;
}
else if(s[j] == '(') flag ++;
else if(s[j] == ')') flag --;
else if(s.substr(j,2) == "if") flag ++;
else if(s.substr(j,4) == "else") flag --;
else if(s[j] == '{') flag ++;
else if(s[j] == '}') {
flag --;
if(flag == 1 && s[ignore_space(s,j + 1)] != 'e') flag --;
if(flag == 0) {
s.insert(j + 1,"}");
break;
}
}
j ++;
}
}
break;
}
}
}
else if(s.substr(i,4) == "else" && !isalpha(s[i - 1]) && !isalpha(s[i + 4])) {
int j = trim_space(s,i + 4);
if(s[j] != '{') {
s.insert(j,"{");
j ++;
int flag = 0;
while(j < s.size()) {
if(s[j] == ';' && flag == 0) {
s.insert(j + 1,"}");
break;
}
else if(s[j] == '(') flag ++;
else if(s[j] == ')') flag --;
else if(s.substr(j,2) == "if") flag ++;
else if(s.substr(j,4) == "else") flag --;
else if(s[j] == '{') flag ++;
else if(s[j] == '}') {
flag --;
if(flag == 1 && s[ignore_space(s,j + 1)] != 'e') flag --;
if(flag == 0) {
s.insert(j + 1,"}");
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 = 0,spa = 0,f = 1;
string s;
getline(cin,s);
perfect(s);
for(int i = ignore_space(s,0);i < s.size();i ++) {
if(f) {
f = 0;
if(s[i] == '}') put_space(spa - 1);
else put_space(spa);
}
if(s[i] == '{') {
int d = i - 1;
while(d >= 0 && s[d] == ' ') d --;
if(d < 0) d ++;
if(s[d] == '=') {
while(s[i] != '}') putchar(s[i ++]);
putchar(s[i]);
continue;
}
if(!flag || !check(s[d])) {
putchar('\n');
put_space(spa);
}
flag = 1;
putchar(s[i]);
putchar('\n');
spa ++;
f = 1;
i = ignore_space(s,i + 1) - 1;
}
else if(s[i] == '}') {
spa --;
putchar(s[i]);
i = ignore_space(s,i + 1) - 1;
if(s[i + 1] == '}') {
putchar('\n');
f = 1;
}
else if(i < s.size() - 1) {
putchar('\n');
f = 1;
}
}
else if(s[i] == '(') {
int d = 1,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 + 1) - 1;
}
else if(s[i] == ';') {
putchar(s[i]);
i = ignore_space(s,i + 1) - 1;
putchar('\n');
f = 1;
}
else {putchar(s[i]);}
}
return 0;
}