L3-019 代码排版 （30 分)

某编程大赛中设计有一个挑战环节，选手可以查看其他选手的代码，发现错误后，提交一组测试数据将对手挑落马下。为了减小被挑战的几率，有些选手会故意将代码写得很难看懂，比如把所有回车去掉，提交所有内容都在一行的程序，令挑战者望而生畏。

为了对付这种选手，现请你编写一个代码排版程序，将写成一行的程序重新排版。当然要写一个完美的排版程序可太难了，这里只简单地要求处理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 ;
}