1112. Stucked Keyboard (20)

题目链接：点击打开链接

问题：字符串中连续出现K次的字符为坏键.

AC思路：用vector记录每个字符出现的位置（遍历一遍字符串），然后根据每个字符出现次数，若不能整除K，则说明该字符合法;

否则依次对比出现的位置pos[i + K - 1] - pos[i] == K - 1，若是，则该字符不合法；否则记为合法。

代码如下：

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <cstring>
#include <cstdio>
#include <map>
using namespace std;

bool vis[256] = {false}; // 记录字符是否合法

vector<int> v[256]; // 记录字符出现的位置
vector<char> order; // 记录每个字符访问的顺序

int main() {
	int n;
	cin >> n;
	string str;
	cin >> str;
	for (int i = 0; i < str.length(); i++) {
		v[str[i]].push_back(i);
		if (vis[str[i]] == false) { // 这里的vis用于防止order重复记录同一个字符
			order.push_back(str[i]);
			vis[str[i]] = true;
		}
	}
	
	memset(vis, false, sizeof(vis));
	for (int i = 0; i < order.size(); i++) {
		int t = order[i];
		if (v[t].size() % n) {
			vis[t] = true;
			continue;
		} else {
			bool flag = false;
			for (int j = 0; j < v[t].size(); j += n) { // 依次对比字符出现的位置是否合法
				if (v[t][j + n - 1] - v[t][j] != n - 1) {
					flag = true;
					break;
				}
			}
			vis[t] = flag;
			if (flag == false) {
				cout << order[i];
			}
		}
	}
	cout << endl;
	
	string ans;
	for (int i = 0; i < str.length(); i++) {
		if (vis[ str[i] ]) {
			ans += str[i];
		} else {
			ans += str[i];
			i += n - 1;
		}
	}
	cout << ans << endl;
    return 0;
}