19. 连续字母长度

题目描述

给定一个字符串，只包含大写字母，求在包含同一字母的子串中，长度第 k 长的子串的长度，相同字母只取最长的那个子串。

输入描述

第一行有一个子串(1<长度<=100)，只包含大写字母。

第二行为 k的值

输出描述

输出连续出现次数第k多的字母的次数。

用例

输入	AAAAHHHBBCDHHHH 3
输出	2
说明	同一字母连续出现的最多的是A和H，四次；第二多的是H，3次，但是H已经存在4个连续的，故不考虑；下个最长子串是BB，所以最终答案应该输出2。

输入	AABAAA 2
输出	1
说明	同一字母连续出现的最多的是A，三次；第二多的还是A，两次，但A已经存在最大连续次数三次，故不考虑；下个最长子串是B，所以输出1。

输入	ABC 4
输出	-1
说明	只含有3个包含同一字母的子串，小于k，输出-1

输入	ABC 2
输出	1
说明	三个子串长度均为1，所以此时k = 1，k=2，k=3这三种情况均输出1。特此说明，避免歧义。

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

int main()
{
    string str;
    cin >> str;
    str += '0';
    int n;
    cin >> n;

    char cur = str[0];
    int count = 1;
    map<char, int>mp;
    for (int i  = 1; i < str.size(); i++) {
        if (cur != str[i]) {
            if (mp.find(cur) == mp.end()) {
                mp[cur] = count;
            } else {
                mp[cur] = max(mp[cur], count);
            }

            cur = str[i];
            count = 1;
        } else {
            count++;
        }
    }

    if (n > mp.size()) {
        cout << -1 << endl;
        return 0;
    }

    vector<pair<char, int>> res(mp.begin(), mp.end());
    sort(res.begin(), res.end(), [](const pair<char, int>& a, const pair<char, int>& b) {
        return a.second > b.second;
    });
    cout << res[n-1].second << endl;
    return 0;
}