【CCCC】L3-019 代码排版 (30分)，大模拟

problem

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;
}

• 给出一行代码,要求排版成整篇
• 缩进2空格，花括号不换行，每行封号结尾

solution

• 去掉main单独输出（括号内可能有空格，WA1），剩下的按照每行的内容可以简单分为3类，遇到花括号（增加或减少缩进），逻辑判断（for,while,if+()一类,else一类），普通语句（以封号；结尾）。
• 处理逻辑判断的时候有2个坑点，第一个坑点是考虑条件()内可能也有()，第二个是单句前后要加上花括号。其他的缩进修改和普通语句都比较简单直接处理就可。

#include<bits/stdc++.h>
using namespace std;

//判断语句块类型
int judge(string dat, int i){
	//WA3:当前位置是if并且不是在字符串内
	if(dat.find("if", i)==i && (dat[i+2]==' '||dat[i+2]=='('))return 2;
	if(dat.find("for",i)==i && (dat[i+3]==' ' ||dat[i+3]=='('))return 3;
    if(dat.find("while",i)==i && (dat[i+5]==' '||dat[i+5]=='('))return 5;
    if(dat.find("else",i)==i && dat[i+4]==' ')return 4;
	return 0;//普通语句
}
//输出前删除多余空格, 并输出当前对应的空格
void erase_space(string dat,int &i){while(dat[i]==' ')i++;}
void print_space(int sp){for(int i=0;i<sp;i++)putchar(' ');}

int main(){
	string dat;  getline(cin,dat);
	
	//处理int main()  找i和)输出
	int l = dat.find('i',0), r = dat.find(')',0);
	cout<<dat.substr(l,r-l+1)<<"\n{\n";
	
	//处理其他，按照行分类
	int tmp, space = 2;//语句类型,空格数
	int flag, debt=0;//单句标记，层数（补全缺少的}）
	for(int i = dat.find('{')+1,j=0,k; i < dat.size(); ){
		erase_space(dat,i);//删除每行前的空格
		if(dat[i]=='{' || dat[i]=='}'){
			if(dat[i]=='{'){
				print_space(space);
				printf("{\n");
				space += 2;
				i++;
				continue;
			}else{
				space -= 2;
				print_space(space);
				printf("}\n");
				i++;
				if(space==0)break;//main的}输完就结束了
				
				//【重复】单句特判
				erase_space(dat,i);
				while(debt && judge(dat,i)!=4){
					space -= 2;
					print_space(space);
					printf("}\n");
					debt--;
				}
			}
		}else if((tmp=judge(dat,i))){
			print_space(space);
			//处理for,while,if,+()或者else
			if(tmp==4){
				printf("else");
				k = i+3;
			}else{
				cout<<dat.substr(i,tmp)<<" ";
				i += tmp;
				erase_space(dat, i);
				//考虑if()中也有()条件的情况
				k = i; int t = 0;
				while(1){
					if(dat[k]=='(')t++;
					if(dat[k]==')')t--;
					if(!t)break;
					k++;
				}
				cout<<dat.substr(i,k-i+1);
			}
			//预处理{}的内容,考虑单句特判
			int m = k+1;
			erase_space(dat,m);
			if(dat[m] != '{'){//单句标记
				printf(" {\n");
				flag = 1;
				debt++;
				i = m;
			}else{
				printf(" {\n");
				flag = 0;
				i = m+1;
			}
			space += 2;
		}else{//普通语句
			int ed = dat.find(';', i);
			print_space(space);
			cout<<dat.substr(i,ed-i+1)<<"\n";
			i = ed+1;
			
			//这是单句内的语句
			if(flag && debt){
				space -= 2;
				print_space(space);
				printf("}\n");
				debt--;
				
				//【重复】单句特判
				erase_space(dat,i);
				while(debt && judge(dat,i)!=4){
					space -= 2;
					print_space(space);
					printf("}\n");
					debt--;
				}
			}
		}
	}
	return 0;
}