任务调度器

任务调度器

问题描述 :
给定一个用字符数组表示的 CPU 需要执行的任务列表。其中包含使用大写的 A - Z 字母表示的26 种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。CPU 在任何一个单位时间内都可以执行一个任务，或者在待命状态。然而，两个相同种类的任务之间必须有长度为 n 的冷却时间，因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。你需要计算完成所有任务所需要的最短时间。

示例 ：
输入：tasks = [“A”,“A”,“A”,“B”,“B”,“B”], n = 2
输出：8
解释：A -> B -> (待命) -> A -> B -> (待命) -> A -> B.
在本示例中，两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间，而执行一个任务只需要一个单位时间，所以中间出现了（待命）状态。

说明：
任务的总个数为 [1, 10000]。
n 的取值范围为 [0, 100]。

输入说明 :
首先输入任务的数目len
然后输入len个大写的 A - Z 字母，无空格、无引号
最后输入n

输出说明 :
输出一个整数，表示结果

输入范例 :
6
AAABBB
2

输出范例：
8

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

bool cmp(int &a, int &b){
    return a > b;
}

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
    	if (!n) return tasks.size();
        vector<int> nums(26, 0);
        for (char task:tasks){
            nums[task-'A']++;
        }
        sort(nums.begin(),nums.end(),cmp);//从大到小排序
        int total=(n+1)*(nums[0]-1)+1;
        for (int i=1;i<nums.size();i++){
            if (nums[i]==nums[0])
                total++;
        } 
		return total>tasks.size()?total:tasks.size();
    }
};

int main(){
    vector<char> v;
    int len,n;
    char data;
    cin>>len;
    for(int i=0; i<len; i++){
        cin>>data;
        v.push_back(data);
    }
    cin>>n;
    int result=Solution().leastInterval(v,n);
    cout<<result<<endl;
    return 0;
}