[1800][坑] CF978B 构造 结论

Author: EndlessCheng

main/900-999/979B.go

@@ -0,0 +1,52 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF979B(_r io.Reader, out io.Writer) {
+	in := bufio.NewReader(_r)
+	var k, ans, ansI, cntAns int
+	var s string
+	Fscan(in, &k)
+	for i := 0; i < 3; i++ {
+		Fscan(in, &s)
+		n := len(s)
+		c := ['z' + 1]int{}
+		for _, b := range s {
+			c[b]++
+		}
+		mx := 0
+		if c[s[0]] < n {
+			for _, c := range c {
+				if c > mx {
+					mx = c
+				}
+			}
+			if mx+k < n {
+				mx += k
+			} else {
+				mx = n
+			}
+		} else if k == 1 && n > 1 {
+			mx = n - 1
+		} else {
+			mx = n
+		}
+		if mx > ans {
+			ans, ansI, cntAns = mx, i, 1
+		} else if mx == ans {
+			cntAns++
+		}
+	}
+	if cntAns > 1 {
+		Fprint(out, "Draw")
+	} else {
+		Fprint(out, []string{"Kuro", "Shiro", "Katie"}[ansI])
+	}
+}
+
+//func main() { CF979B(os.Stdin, os.Stdout) }

main/900-999/979B_test.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/979/B
+// https://codeforces.com/problemset/status/979/problem/B
+func TestCF979B(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+3
+Kuroo
+Shiro
+Katie
+outputCopy
+Kuro
+inputCopy
+7
+treasurehunt
+threefriends
+hiCodeforces
+outputCopy
+Shiro
+inputCopy
+1
+abcabc
+cbabac
+ababca
+outputCopy
+Katie
+inputCopy
+15
+foPaErcvJ
+mZaxowpbt
+mkuOlaHRE
+outputCopy
+Draw
+inputCopy
+60
+ddcZYXYbZbcXYcZdYbddaddYaZYZdaZdZZdXaaYdaZZZaXZXXaaZbb
+dcdXcYbcaXYaXYcacYabYcbZYdacaYbYdXaccYXZZZdYbbYdcZZZbY
+XaZXbbdcXaadcYdYYcbZdcaXaYZabbXZZYbYbcXbaXabcXbXadbZYZ
+outputCopy
+Draw`
+	testutil.AssertEqualCase(t, rawText, -1, CF979B)
+}