华为题：任务调度

给定一个字符数组 tasks，表示 CPU 需要执行的任务列表。每个任务可以用一个大写字母表示。任务可以按任意顺序执行，且每个任务可以在 1 个单位时间内执行完。

两个相同种类的任务之间必须有长度为 n 的冷却时间。请计算完成所有任务所需的最短时间。

输入描述

第一行输入一个字符串 tasks，表示任务列表。

第二行输入一个整数 n，表示冷却时间。

输出描述

输出完成所有任务所需的最短时间。

示例 1

输入

text

AAABBB
2

输出

text

8

说明

一种可行安排：A -> B -> idle -> A -> B -> idle -> A -> B，共 8 个单位时间。

C++ 解法

cpp

#include <bits/stdc++.h>
using namespace std;

int main() {
    string tasks;
    cin >> tasks;

    int n;
    cin >> n;

    vector<int> cnt(26, 0);
    for (char c : tasks) cnt[c - 'A']++;

    sort(cnt.begin(), cnt.end(), greater<int>());

    int maxCount = cnt[0];
    int maxCountNum = 0;

    for (int x : cnt) {
        if (x == maxCount) maxCountNum++;
    }

    int ans = (maxCount - 1) * (n + 1) + maxCountNum;

    cout << max(ans, (int)tasks.size()) << endl;
    return 0;
}