PAT 1112. Stucked Keyboard (20) 寻找坏掉的按键（重复按键）

/*************************
题意:
键盘上有一些按键坏了
导致一旦按下这个按键，则“一定”会输出k次
给出一个字符串，求这个字符串中坏掉的按键，
并给出修正后的字符串
并按“出现顺序”输出那些错误的按键。
************************/
/***********************
解题思路：
关键在于坏掉的按键“一定”会输出k次。
故其连续出现的次数n一定能被k整除
例如k=3，但是你s键连续出现了5次，则s键不是坏键。
注意点：
	1. 如果一个键检测到发现不是坏键，则后续就一定不是坏键。
	2. 用for循环判断连续个数时，勿漏for结束后的处理  
	3. 输出错误按键的顺序是按出现的次序。

*************************/
/***********************

*********************/
#include<iostream>
#include<stdio.h>
#include<string>
#include<vector>
#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define M 150
#define INF 0x7ffffff

struct People{
	string name;
	int high;
};

int firstp[M];
bool cmp(int a, int b){
	return firstp[a] < firstp[b];
}

vector<People> v;
People num[M];
int main(){
	int k, i, n;
	string s;
	cin>>k;
	cin>>s;
	char lastc=0;
	string ans;
	int flag[150];
	for(i = 0;i<150;i++)
		flag[i] = -1;
	lastc = 0;
	n = 1;
	for(i = 0;i<s.size();i++){
		if(i == 0 || s[i] != lastc){
			if(flag[lastc] == -1){
				firstp[lastc] = i - n;
			}
			if(n%k != 0)
				flag[lastc] = 0;
			else if(flag[lastc] != 0) //当等于0，就定型了。 
				flag[lastc]  = 1;
			lastc = s[i];
			n = 1;
		}
		else
			n++;
	}
	
	if(flag[lastc] == -1)
		firstp[lastc] = i - n;
	if(n%k != 0)
		flag[lastc] = 0;		
	else if(flag[lastc] != 0) //当等于0，就定型了。 
		flag[lastc]  = 1;


	vector<int> dc;
	for(i = 0;i<M;i++)
		if(flag[i] == 1)
			dc.push_back(i);
	sort(dc.begin(),dc.end(),cmp);
	for(i=0;i<dc.size();i++)
		cout<<(char)dc[i];
	cout<<endl;

	for(i = 0;i<s.size();i++){
		cout<<s[i];
		if(flag[s[i]] == 1){
			i += k-1;
		}
	}
	cout<<endl;
	return 0;
}