PAT(basic level) 1033 旧键盘打字(20)

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

输入格式：

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

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

输出格式：

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

输入样例：

7+IE.
7_This_is_a_test.

输出样例：

_hs_s_a_tst

#include <iostream>
using namespace std;

int main()
{
	int num[10] = {0};   //数字 
	int a[26] = {0};    //小写字母 
	int A[26] = {0};    //大写字母 
	int k0 = 0, k1 = 0, k2 = 0, k3 = 0;   //标识 '_', ',', '.', '-' 
	string bad, s;
	getline(cin, bad);   //按行输入，*可能没有坏的键 
	getline(cin, s);
	for(int i = 0; i < bad.size(); i ++){   //遍历坏的键，将不能输出的置1 
		if(bad[i] >= '0' && bad[i] <= '9') {num[bad[i]-'0'] = 1; }
		if(bad[i] >= 'A' && bad[i] <= 'Z') {A[bad[i]-'A'] = 1;a[bad[i]-'A'] = 1;   }
		if(bad[i] == '_') k0 = 1;
		if(bad[i] == ',') k1 = 1;
		if(bad[i] == '.') k2 = 1;
		if(bad[i] == '-') k3 = 1;
		if(bad[i] == '+') {
			for(int j = 0; j < 26; j ++)
				A[j] = 1;
		}
	}
	string s1;
	for(int i = 0; i < s.size(); i ++){    
		if(s[i] >= '0' && s[i] <= '9') {if(num[s[i]-'0']) continue; else s1+=s[i];}
		if(s[i] >= 'a' && s[i] <= 'z') {if(a[s[i]-'a']) continue; else s1+=s[i];}
		if(s[i] >= 'A' && s[i] <= 'Z') {if(A[s[i]-'A']) continue; else s1+=s[i];}
		if(s[i] == '_') {if(k0) continue; else s1 += s[i];}
		if(s[i] == ',') {if(k1) continue; else s1 += s[i];}
		if(s[i] == '.') {if(k2) continue; else s1 += s[i];}
		if(s[i] == '-') {if(k3) continue; else s1 += s[i];}
	}
	cout << s1;
	return 0;
}