621.任务调度器

62 阅读1分钟

题目:
给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行,并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间,CPU 可以完成一个任务,或者处于待命状态。

然而,两个 相同种类 的任务之间必须有长度为整数 ****n ****的冷却时间,因此至少有连续 n 个单位时间内 CPU 在执行不同的任务,或者在待命状态。

你需要计算完成所有任务所需要的 最短时间 。
算法:
官方题解的解法一: leetcode.cn/problems/ta…

func leastInterval(tasks []byte, n int) int {
	taskCount := make([]int, 26)
	nextTime := make([]int, 26)
	for i := range tasks {
		idx := int(tasks[i] - 'A')
		nextTime[idx] = 1
		taskCount[idx] ++
	}

	time := 0
	for i := 0; i < len(tasks); i ++ {
		// 找到最小的next time
		time ++
		minNextTime := math.MaxInt64
		for k := range taskCount {
			if taskCount[k] > 0 && nextTime[k] < minNextTime {
				minNextTime = nextTime[k]
			}
		}
		if minNextTime > time {
			time = minNextTime
		}
		// 找到调度次数最多的任务的index
		maxCountTask := -1
		for j := range taskCount {
			if taskCount[j] > 0 && nextTime[j] <= time &&(maxCountTask == -1 || taskCount[j] > taskCount[maxCountTask]) {
				maxCountTask = j
			}
		}
		// 调度该任务,更新time
		// task次数--
		nextTime[maxCountTask] = time + n + 1
		taskCount[maxCountTask] --
	}
	
	return time
}