Merge remote-tracking branch 'origin/master'

Author: EndlessCheng

copypasta/template/interactive_problem/README.md

@@ -1,6 +1,6 @@
 # Interactive Problem Template
 
-The idea is design an interface below, then implement it both in `main.go` and `main_test.go`.
+Use below interface and implement it both in `main.go` and `main_test.go`.
 
 ```go
 type interaction interface {
@@ -10,4 +10,4 @@ type interaction interface {
 }
 ``` 
 
-In this way we can mock the IO part in `main_test.go`.
+In this way we can mock the IO part.

main/1500-1599/1593F.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1593F(in io.Reader, out io.Writer) {
+	abs := func(x int) int {
+		if x < 0 {
+			return -x
+		}
+		return x
+	}
+
+	var T, n, ma, mb int
+	var s string
+	for Fscan(in, &T); T > 0; T-- {
+		Fscan(in, &n, &ma, &mb, &s)
+		minD, ans := n, "-1"
+		t := make([]byte, n)
+		vis := [40][40][40][41]bool{}
+		var f func(i, x, y, c int)
+		f = func(i, x, y, c int) {
+			if i == n {
+				if x == 0 && y == 0 && abs(n-c*2) < minD {
+					minD = abs(n - c*2)
+					ans = string(t)
+				}
+				return
+			}
+			if vis[i][x][y][c] {
+				return
+			}
+			vis[i][x][y][c] = true
+			t[i] = 'R'
+			f(i+1, (x*10+int(s[i]&15))%ma, y, c+1)
+			t[i] = 'B'
+			f(i+1, x, (y*10+int(s[i]&15))%mb, c)
+		}
+		f(0, 0, 0, 0)
+		Fprintln(out, ans)
+	}
+}
+
+//func main() { CF1593F(os.Stdin, os.Stdout) }

main/1500-1599/1593F_test.go

@@ -0,0 +1,29 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1593/F
+// https://codeforces.com/problemset/status/1593/problem/F
+func TestCF1593F(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+4
+5 3 13
+02165
+4 2 1
+1357
+8 1 1
+12345678
+2 7 9
+90
+outputCopy
+RBRBR
+-1
+BBRRRRBB
+BR`
+	testutil.AssertEqualCase(t, rawText, 0, CF1593F)
+}

main/1500-1599/1593G.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1593G(_r io.Reader, _w io.Writer) {
+	in := bufio.NewReader(_r)
+	out := bufio.NewWriter(_w)
+	defer out.Flush()
+	abs := func(x int) int {
+		if x < 0 {
+			return -x
+		}
+		return x
+	}
+
+	var T, q, l, r int
+	var s string
+	for Fscan(in, &T); T > 0; T-- {
+		Fscan(in, &s, &q)
+		sum := make([]int, len(s)+1)
+		for i, b := range s {
+			sum[i+1] = sum[i]
+			if b < 42 {
+				sum[i+1] += i&1<<1 - 1
+			}
+		}
+		for ; q > 0; q-- {
+			Fscan(in, &l, &r)
+			Fprintln(out, abs(sum[r]-sum[l-1]))
+		}
+	}
+}
+
+//func main() { CF1593G(os.Stdin, os.Stdout) }

main/1500-1599/1593G_test.go

@@ -0,0 +1,35 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1593/G
+// https://codeforces.com/problemset/status/1593/problem/G
+func TestCF1593G(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+3
+([))[)()][]]
+3
+1 12
+4 9
+3 6
+))))))
+2
+2 3
+1 4
+[]
+1
+1 2
+outputCopy
+0
+2
+1
+0
+0
+0`
+	testutil.AssertEqualCase(t, rawText, 0, CF1593G)
+}

main/1600-1699/1609D.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+	"sort"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1609D(_r io.Reader, _w io.Writer) {
+	in := bufio.NewReader(_r)
+	out := bufio.NewWriter(_w)
+	defer out.Flush()
+
+	var n, d, v, w, ex int
+	Fscan(in, &n, &d)
+	fa := make([]int, n+1)
+	sz := make([]int, n+1)
+	for i := range fa {
+		fa[i] = i
+		sz[i] = 1
+	}
+	var find func(int) int
+	find = func(x int) int {
+		if fa[x] != x {
+			fa[x] = find(fa[x])
+		}
+		return fa[x]
+	}
+
+	for ; d > 0; d-- {
+		Fscan(in, &v, &w)
+		v, w = find(v), find(w)
+		if v == w {
+			ex++
+		} else {
+			if sz[v] > sz[w] {
+				v, w = w, v
+			}
+			sz[w] += sz[v]
+			sz[v] = 0 // 注意这里
+			fa[v] = w
+		}
+		a := append([]int(nil), sz...)
+		sort.Ints(a) // or kth
+		s := -1
+		for i := n; i >= n-ex; i-- {
+			s += a[i]
+		}
+		Fprintln(out, s)
+	}
+}
+
+//func main() { CF1609D(os.Stdin, os.Stdout) }

main/1600-1699/1609D_test.go

@@ -0,0 +1,48 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1609/D
+// https://codeforces.com/problemset/status/1609/problem/D
+func TestCF1609D(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+7 6
+1 2
+3 4
+2 4
+7 6
+6 5
+1 7
+outputCopy
+1
+1
+3
+3
+3
+6
+inputCopy
+10 8
+1 2
+2 3
+3 4
+1 4
+6 7
+8 9
+8 10
+1 4
+outputCopy
+1
+2
+3
+4
+5
+5
+6
+8`
+	testutil.AssertEqualCase(t, rawText, -1, CF1609D)
+}

main/1600-1699/1611D.go

@@ -0,0 +1,50 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1611D(_r io.Reader, _w io.Writer) {
+	in := bufio.NewReader(_r)
+	out := bufio.NewWriter(_w)
+	defer out.Flush()
+
+	var T, n, root, v int
+	for Fscan(in, &T); T > 0; T-- {
+		Fscan(in, &n)
+		pa := make([]int, n)
+		for i := range pa {
+			Fscan(in, &pa[i])
+			if pa[i]--; pa[i] == i {
+				root = i
+			}
+		}
+		Fscan(in, &v)
+		ok := v-1 == root
+		dep := make([]int, n)
+		for i := 1; i < n; i++ {
+			Fscan(in, &v)
+			if !ok {
+				continue
+			}
+			v--
+			if pa[v] != root && dep[pa[v]] == 0 {
+				ok = false
+			}
+			dep[v] = i
+		}
+		if ok {
+			for v, d := range dep {
+				Fprint(out, d-dep[pa[v]], " ")
+			}
+			Fprintln(out)
+		} else {
+			Fprintln(out, -1)
+		}
+	}
+}
+
+//func main() { CF1611D(os.Stdin, os.Stdout) }

main/1600-1699/1611D_test.go

@@ -0,0 +1,40 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1611/D
+// https://codeforces.com/problemset/status/1611/problem/D
+func TestCF1611D(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+4
+5
+3 1 3 3 1
+3 1 2 5 4
+3
+1 1 2
+3 1 2
+7
+1 1 2 3 4 5 6
+1 2 3 4 5 6 7
+6
+4 4 4 4 1 1
+4 2 1 5 6 3
+outputCopy
+1 10 0 102 100
+-1
+0 3 100 1 1 2 4
+6 5 10 0 2 3
+inputCopy
+1
+9
+2 2 2 2 2 2 2 2 2
+1 2 6 7 3 9 5 8 4
+outputCopy
+-1`
+	testutil.AssertEqualCase(t, rawText, -1, CF1611D)
+}