upd

main/edu/2/zfunction/step4/c/main.go

@@ -7,44 +7,34 @@ import (
 	"os"
 )
 
-// github.com/EndlessCheng/codeforces-go
+// https://space.bilibili.com/206214
 func run(_r io.Reader, _w io.Writer) {
 	in := bufio.NewReader(_r)
 	out := bufio.NewWriter(_w)
 	defer out.Flush()
-	min := func(a, b int) int {
-		if a < b {
-			return a
-		}
-		return b
-	}
-	max := func(a, b int) int {
-		if a > b {
-			return a
-		}
-		return b
-	}
 
 	var T int
 	var s []byte
 	for Fscan(in, &T); T > 0; T-- {
 		Fscan(in, &s)
 		n := len(s)
 		z := make([]int, n)
-		d := make([]int, n)
-		for i, l, r := 1, 0, 0; i < n; i++ {
-			z[i] = max(0, min(z[i-l], r-i+1))
+		diff := make([]int, n)
+		l, r := 0, 0
+		for i := 1; i < n; i++ {
+			if i <= r {
+				z[i] = min(z[i-l], r-i+1)
+			}
 			for i+z[i] < n && s[z[i]] == s[i+z[i]] {
 				l, r = i, i+z[i]
 				z[i]++
 			}
-			d[0]++
-			d[z[i]]--
+			diff[z[i]]--
 		}
-		c := 1
-		for _, v := range d {
-			c += v
-			Fprint(out, c, " ")
+		sumD := n
+		for _, d := range diff {
+			sumD += d
+			Fprint(out, sumD, " ")
 		}
 		Fprintln(out)
 	}