621 accepted, 19ms, the fast is 9ms

Author: aQua

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

@@ -12,24 +12,31 @@ func leastInterval(tasks []byte, n int) int {
 	sort.Ints(rec)
 
 	res := 0
-	remain := 0
-	begin := 0
+	intervals := 0
+	idx := 0
 	for len(rec) > 0 {
-		if remain > 0 {
-			res += remain
+		if intervals > 0 {
+			res += intervals
 		}
 
-		remain = n
+		// 新周期
+		intervals = n + 1
 
-		begin = len(rec) - 1
-		for begin >= 0 && rec[begin] > 0 {
-			rec[begin]--
-			remain--
-			begin--
+		idx = len(rec) - 1
+		for intervals > 0 && idx >= 0 && rec[idx] > 0 {
+			rec[idx]--
+			intervals--
+			idx--
 			res++
 		}
 
-		rec = rec[begin+1:]
+		sort.Ints(rec)
+		idx = len(rec) - 1
+		for idx >= 0 && rec[idx] > 0 {
+			idx--
+		}
+
+		rec = rec[idx+1:]
 	}
 
 	return res

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

@@ -47,6 +47,16 @@ func Test_Problem0621(t *testing.T) {
 				12,
 			},
 		},
+
+		question{
+			para{
+				[]byte{'A', 'A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'},
+				100,
+			},
+			ans{
+				506,
+			},
+		},
 		// 如需多个测试，可以复制上方元素。
 	}