PAT (Advanced Level) Practice - 1112 Stucked Keyboard（20 分）

题目链接：​​点击打开链接​​

题目大意：键盘某些键卡住了，按一次重复 n(题目当中的 k) 次，要求找出可能的键，并且输出正确的字符串顺序。可能的键要求按照被发现的顺序输出。

解题思路：

1. 如果 s[i]==s[i-1]，则 cnt++，否则就 cnt%n == 0？若是，则用 vector 记录可能是坏键，为什么可能是，因为有可能一开始是坏键情况，但是最后又不是坏键盘情况，比如：“3 aaasaa”，“a”就不算是坏键。set 用来去重。而unordered_map 用来标记一定不是坏键的键。
2. 最后输出若是坏键，则输出一个，然后 i+=n-1；若不是坏键，则直接输出。

AC 代码

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a);
#define INF 0x3f3f3f3f
#define MOD 1000000007

using namespace std;

typedef long long ll;

unordered_map<char,int> ump;
vector<char> v;
set<char> st;

int main()
{
    int n,idx; scanf("%d",&n);
    string s; cin>>s;
    char c;
    int len=s.length(),cnt=1;
    for(int i=1;i<len;i++)
    {
        if(s[i]==s[i-1]) cnt++;
        else
        {
            if(cnt%n==0)
            {
                if(st.insert(s[i-1]).second)
                    v.push_back(s[i-1]);
            }
            else ump[s[i-1]]=-1;

            cnt=1;
        }
    }

    if(cnt%n==0)
    {
        if(st.insert(s[len-1-1]).second)
            v.push_back(s[len-1-1]);
    }
    else ump[s[len-1-1]]=-1;


    for(int i=0;i<v.size();i++)
        if(ump[v[i]]!=-1) putchar(v[i]);
    puts("");

    for(int i=0;i<len;i++)
    {
        c=s[i];
        if(ump[c]==-1) putchar(c);
        else
        {
            putchar(c);
            i+=n-1;
        }
    }
    puts("");

    return 0;
}