1112 Stucked Keyboard （20 point(s)）

题解

加边界条件减少判断。。。此题在字符串尾部加“#”。

通过改变输入来减少考虑情况。 类似manacher算法在首部加$符号、尾部加\0.

#include<iostream>
#include<map>
#include<cstdio>
#include<vector>
using namespace std;
int k, cnt = 1;
map<char, bool> sure, havep;
vector<char> stuck;
string t;
char pre = '#';
int main() {
	scanf("%d", &k);
	cin >> t;
	t += "#"; // 作为边界条件 
	for(int i = 0; i < t.length(); ++i) {
		if(pre != t[i]) {
			if(!sure[pre]) sure[pre] = cnt % k; 
			cnt = 0;
		}
		pre = t[i];
		++cnt;
	}
	for(int i = 0; i < t.length() - 1; ++i) if(!sure[t[i]] && !havep[t[i]]) {
		havep[t[i]] = true;
		printf("%c", t[i]);
	}
	printf("\n");
	for(int i = 0; i < t.length() - 1; sure[t[i]] ? ++i : i += k) printf("%c", t[i]);
	return 0;
}