1593:Alignment of Code

Alignment of Code

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<sstream>
using namespace std;
const int maxw = 180 + 5;
const int maxl = 1000 + 5;
int space[maxw];
vector<string>code[maxl];
void prspa(int t){
    while(t--) cout<<' ';
}
int main(){
    string line,s;
    int cnt = 0;
    while(getline(cin,line)){
        int cnt2 = 0;
        stringstream ss(line);
        while(ss >> s){
            code[cnt].push_back(s);
            space[cnt2] = max(space[cnt2],(int)s.length());
            cnt2++;
        }
        cnt++;
    }
    // 这样写老报错，还是用sstream吧。。。
    // while(true){
    //     int cnt2 = 0;
    //     while(cin>>s && s[0] != '\n'){
    //         code[cnt].push_back(s);
    //         space[cnt2] = max(space[cnt2],(int)s.length());
    //         cnt2++;
    //     }
    //     cnt++;
    //     if(!cnt2) break;
    // }
    for(int i = 0;i < cnt;i++){
        for(int j = 0;j < code[i].size()-1;j++){
            cout<<code[i][j];
            prspa(space[j]+1 - code[i][j].length());
        }
        cout<<code[i][code[i].size()-1]<<endl; //最后一列不能有空格
    }
    return 0;
}