PAT 1033 旧键盘打字

PAT 1033 旧键盘打字

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10; 
bool f[N];  // false 表示没坏，true 表示坏了 

int main() {

	string s;
	// 题目保证第二行输入的字符串非空，也就意味着第一行输入的字符串有可能会是空的
	// 所以用 cin 输入是不行的（这是个坑啊。。。） 
	getline(cin, s);   
	bool flag = false;  // 代表加号 + 还没有坏 
	for (int i = 0; i < s.size(); i++) {
		char c = s[i];
		f[c] = true;
		if (c == '+') flag = true;
		if (c >= 'A' && c <= 'Z') { // 大写字母
			f[c + 32] = true;
		} 
	}
	string s1;
	cin >> s1;
	bool is_cout = false;
	for (int i = 0; i < s1.size(); i++) {
		char c = s1[i];
		if (!f[c]) {
			if (c >= 'A' && c <= 'Z') {
				if (!flag) {
					cout << c;
					is_cout = true;
				}
			} else {
				cout << c;
				is_cout = true;
			}
		}
	}
	if (!is_cout) cout << endl;

	return 0;
}