CF771B

Author: EndlessCheng

main/700-799/771B.go

@@ -0,0 +1,29 @@
+package main
+
+import (
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF771B(in io.Reader, out io.Writer) {
+	var n, k int
+	var s string
+	Fscan(in, &n, &k)
+	a := make([][]byte, n)
+	for i := 0; i < k-1; i++ {
+		a[i] = []byte{'A' + byte(i/26), 'a' + byte(i%26)}
+	}
+	for i := k - 1; i < n; i++ {
+		if Fscan(in, &s); s[0] == 'Y' {
+			a[i] = []byte{'A' + byte(i/26), 'a' + byte(i%26)}
+		} else {
+			a[i] = a[i-k+1]
+		}
+	}
+	for _, v := range a {
+		Fprintf(out, "%s ", v)
+	}
+}
+
+//func main() { CF771B(os.Stdin, os.Stdout) }

main/700-799/771B_test.go

@@ -0,0 +1,29 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/771/B
+// https://codeforces.com/problemset/status/771/problem/B
+func TestCF771B(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+8 3
+NO NO YES YES YES NO
+outputCopy
+Adam Bob Bob Cpqepqwer Limak Adam Bob Adam
+inputCopy
+9 8
+YES NO
+outputCopy
+R Q Ccccccccc Ccocc Ccc So Strong Samples Ccc
+inputCopy
+3 2
+NO NO
+outputCopy
+Na Na Na`
+	testutil.AssertEqualCase(t, rawText, 0, CF771B)
+}