PAT乙级 1033 旧键盘打字

题目：

旧键盘上坏了几个键，于是在敲一段文字的时候，对应的字符就不会出现。现在给出应该输入的一段文字、以及坏掉的那些键，打出的结果文字会是怎样？

输入格式：

输入在 2 行中分别给出坏掉的那些键、以及应该输入的文字。其中对应英文字母的坏键以大写给出；每段文字是不超过 105 个字符的串。可用的字符包括字母 [a-z, A-Z]、数字 0-9、以及下划线 _（代表空格）、,、.、-、+（代表上档键）。题目保证第 2 行输入的文字串非空。

注意：如果上档键坏掉了，那么大写的英文字母无法被打出。

输出格式：

在一行中输出能够被打出的结果文字。如果没有一个字符能被打出，则输出空行。

输入样例：

7+IE.
7_This_is_a_test.

输出样例：

_hs_s_a_tst

---

代码：

#include<iostream>
#include<bitset>
using namespace std;
bitset<130> vis;
int main()
{
	//一开始用 ans = "",ans = ans + b[i] 记录答案，巨慢 
	string a,b;	
	getline(cin,a);//防止第一行是空串 
	cin>>b;
	bool s = true;
	int i=0;
	while(a[i]!='\0')
	{
		if('A'<=a[i]&&a[i]<='Z')//如果是大写字母，那么小写字母也不能输入 
		{
			vis[int(a[i])] = true;
			vis[a[i]-'A'+'a'] = true;
		}
		else
			vis[int(a[i])] = true;
		i++;
	}
	if(vis[int('+')]==true)//上档键坏了
		s = false;
	i = 0;
	while(b[i]!='\0')
	{
		if('A'<=b[i]&&b[i]<='Z')
		{
			if(s==true&&vis[int(b[i])]==false)
				cout<<b[i];
		}
		else
		{
			if(vis[int(b[i])]==false)
				cout<<b[i];
		}
		i++;
	}
	cout<<'\n';
}