1112 Stucked Keyboard

#include<iostream>
#include<map>
#include<set>
#include<string>
using namespace std;
bool sureNobroken[256];
int main(){
int k,cnt=1;
cin>>k;//字符出现次数的阈值
string s;
cin>>s;

map<char,bool>m;//记录是否为坏键
set<char>printed;//记录已经输出的字符

char pre='#';
s=s+'#';
//第一次遍历，标记可能坏掉的键
for(int i=0;i<s.length();i++){
if(s[i]==pre){
cnt++;
}else{
//如果连续出现的不是k的倍数，说明不是坏键
if(cnt%k!=0){
sureNobroken[pre]=true;
}
cnt=1;
}
//如果当前出现的字符数是k的倍数，表示其为坏键
if(i!=s.length()-1){
m[s[i]]=(cnt%k==0);
}
pre=s[i];//更新前一个字符
}
//第二次遍历，修正标记
for(int i=0;i<s.length()-1;i++){
if(sureNobroken[s[i]]){
m[s[i]]=false;
}
}
//输出所有坏键
for(int i=0;i<s.length()-1;i++){
if(m[s[i]]&&printed.find(s[i])==printed.end()){
cout<<s[i];
printed.insert(s[i]);
}
}
cout<<endl;
//输出修正后的字符
for(int i=0;i<s.length()-1;i++){
cout<<s[i];
if(m[s[i]]){
i=i+k-1;
}
}
return 0;
}