cf1943D2

Author: EndlessCheng

main/1900-1999/1943D2.go

@@ -0,0 +1,33 @@
+package main
+
+import (
+	. "fmt"
+	"io"
+)
+
+// https://github.com/EndlessCheng
+func cf1943D2(in io.Reader, out io.Writer) {
+	var T, n, mx, mod int
+	for Fscan(in, &T); T > 0; T-- {
+		Fscan(in, &n, &mx, &mod)
+		f := make([][]int, n+3)
+		for i := range f {
+			f[i] = make([]int, mx+1)
+		}
+		f[1][0] = 1
+		preS := 1
+		for i := 2; i < n+3; i++ {
+			var s, ss, t int
+			for j := mx; j >= 0; j-- {
+				f[i][j] = (preS - ss) % mod
+				t += f[i][j]
+				s += f[i-2][mx-j]
+				ss += s
+			}
+			preS = t
+		}
+		Fprintln(out, (f[n+2][0]+mod)%mod)
+	}
+}
+
+//func main() { cf1943D2(bufio.NewReader(os.Stdin), os.Stdout) }

main/1900-1999/1943D2.txt

@@ -0,0 +1,3 @@
+s[mx] = 0
+s[mx-1] = s[mx] + f[0]
+s[j] = s[j+1] + (f[0] + ... + f[mx - j - 1]) 

main/1900-1999/1943D2_test.go

@@ -0,0 +1,26 @@
+// Generated by copypasta/template/generator_test.go
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1943/D2
+// https://codeforces.com/problemset/status/1943/problem/D2?friends=on
+func Test_cf1943D2(t *testing.T) {
+	testCases := [][2]string{
+		{
+			`4
+3 1 998244853
+4 1 998244353
+3 2 998244353
+343 343 998244353`,
+			`4
+7
+10
+456615865`,
+		},
+	}
+	testutil.AssertEqualStringCase(t, testCases, 0, cf1943D2)
+}