621 finish

Author: aQua

Algorithms/0621.task-scheduler/task-scheduler.go

@@ -1,43 +1,50 @@
 package Problem0621
 
-import (
-	"sort"
-)
-
 func leastInterval(tasks []byte, n int) int {
-	rec := make([]int, 26)
-	for _, b := range tasks {
-		rec[b-'A']++
+	if n == 0 {
+		return len(tasks)
 	}
-	sort.Ints(rec)
-
-	res := 0
-	intervals := 0
-	idx := 0
-	for len(rec) > 0 {
-		if intervals > 0 {
-			res += intervals
-		}
 
-		// 新周期
-		intervals = n + 1
+	rec := [26]int{}
+	for _, c := range tasks {
+		rec[c-'A']++
+	}
 
-		idx = len(rec) - 1
-		for intervals > 0 && idx >= 0 && rec[idx] > 0 {
-			rec[idx]--
-			intervals--
-			idx--
-			res++
+	// 找到最多的任务数
+	most := 0
+	for _, num := range rec {
+		if most < num {
+			most = num
 		}
+	}
 
-		sort.Ints(rec)
-		idx = len(rec) - 1
-		for idx >= 0 && rec[idx] > 0 {
-			idx--
-		}
+	// 整个过程，至少有 idles 个空档需要被填充
+	idles := (most - 1) * (n + 1)
+
+	// 使用每一个 task 去填充 idles
+	for _, num := range rec {
+		// 可以假想 idles+1 个空档，被 most 的 task 等间距n填充后，
+		// 分割成了 most-1 个包含连续 n 个空档的区间
+		// 然后，同样的任务，只能往每个区间中去填充。
+		idles -= min(most-1, num)
+		// 由于是使用 most 做的分割
+		// 如果最后的 idles < 0 只能说明
+		// len(tasks) > (most - 1) * (n + 1)+1
+	}
+
+	return len(tasks) + max(0, idles)
+}
 
-		rec = rec[idx+1:]
+func min(a, b int) int {
+	if a < b {
+		return a
 	}
+	return b
+}
 
-	return res
+func max(a, b int) int {
+	if a > b {
+		return a
+	}
+	return b
 }

Algorithms/0621.task-scheduler/task-scheduler_test.go

@@ -48,6 +48,16 @@ func Test_Problem0621(t *testing.T) {
 			},
 		},
 
+		question{
+			para{
+				[]byte{'A', 'A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'},
+				0,
+			},
+			ans{
+				12,
+			},
+		},
+
 		question{
 			para{
 				[]byte{'A', 'A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'},