PTA --- 天梯赛 L1-064 估值一亿的AI核心代码

6
Hello ?
 Good to chat   with you
can   you speak Chinese?
Really?
Could you show me 5
What Is this prime? I,don ‘t know

Hello ?
AI: hello!
 Good to chat   with you
AI: good to chat with you
can   you speak Chinese?
AI: I can speak chinese!
Really?
AI: really!
Could you show me 5
AI: I could show you 5
What Is this prime? I,don ‘t know
AI: what Is this prime! you,don‘t know

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n;
    cin >> n;
    getchar();
    string s;
    while(n--){
        string str[1005];
        int cnt=0;
        getline(cin,s);
        cout << s << endl << "AI:";
        for(int i=0;i<s.size();++i){
            if(isalnum(s[i])){
                if(s[i]!=‘I‘)
                    s[i]=tolower(s[i]);
            }
            else{
                s.insert(i," ");    //对每个非字母和数字之前加空格
                i++;
            }
            if(s[i]==‘?‘)
                s[i]=‘!‘;
        }
        stringstream ss(s);        //主要依靠这个函数来取消单词之前的空格
        while(ss >> s){
            str[cnt++]=s;
        }
        if(!isalnum(str[0][0]))             // 如果 第一个单词的第一个位置就是标点  那么要输出一个空格，题中要求 每个 AI 的回答前要加上 AI: 和一个空格。
            cout << " ";
        for(int i=0;i<cnt;++i){
            if(!isalnum(str[i][0])){         // 判断第一个字符是否是 标点，标点前面不要输出空格
                cout << str[i];
            }
            else if(str[i]=="can" && str[i+1]=="you"){
                cout << " I can";
                i++;
            }
            else if(str[i]=="could" && str[i+1]=="you"){
                cout << " I could";
                i++;
            }
            else if(str[i]=="I" || str[i]=="me"){
                cout << " you";
            }
            else
                cout << " " << str[i];
        }
        cout << endl;
    }
}