PAT乙级1033 旧键盘打字

这里用cin有一个点过不了，改用getline就没问题了。原因似乎是坏掉的键数有可能为0。

#include <iostream>#include <string>#include <set> using namespace std;int main(){    string str1,str2,str3;    int flag[100000]={0},count=0;    getline(cin,str1);    getline(cin,str2);    set<char> str1Set;    for(int i=0;i<str1.length();i++){        if(str1[i]>='A'&&str1[i]<='Z'){            str1Set.insert(str1[i]+32);        }        str1Set.insert(str1[i]);    }    for(int i=0;i<str2.length();i++){        if(str1Set.count(str2[i])!=0){            flag[i]=1;        }        if(str1Set.count('+')!=0){            if(str2[i]>='A'&&str2[i]<='Z'){                flag[i]=1;            }        }    }    for(int i=0;i<str2.length();i++){        if(flag[i]==1){            continue;        }else{            cout<<str2[i];            count++;        }    }    return 0;}