cf2053B

Author: EndlessCheng

main/2000-2099/2053B.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+"bytes"
+. "fmt"
+"io"
+)
+
+// https://github.com/EndlessCheng
+func cf2053B(in io.Reader, out io.Writer) {
+var T, n int
+for Fscan(in, &T); T > 0; T-- {
+Fscan(in, &n)
+type pair struct{ l, r int }
+a := make([]pair, n)
+cnt := make([]int, n*2+1)
+for i := range a {
+Fscan(in, &a[i].l, &a[i].r)
+if a[i].l == a[i].r {
+cnt[a[i].l]++
+}
+}
+
+// 优化前 https://codeforces.com/problemset/submission/2053/350549792
+s := make([]int, n*2+1)
+for i := 1; i <= n*2; i++ {
+s[i] = s[i-1]
+if cnt[i] == 0 {
+s[i]--
+}
+}
+
+ans := bytes.Repeat([]byte{'0'}, n)
+for i, p := range a {
+if p.l < p.r {
+if s[p.r] < s[p.l-1] {
+ans[i] = '1'
+}
+} else if cnt[p.l] == 1 {
+ans[i] = '1'
+}
+}
+Fprintf(out, "%s\n", ans)
+}
+}
+
+//func main() { cf2053B(bufio.NewReader(os.Stdin), os.Stdout) }

main/2000-2099/2053B_test.go

@@ -0,0 +1,50 @@
+// Generated by copypasta/template/generator_test.go
+package main
+
+import (
+"github.com/EndlessCheng/codeforces-go/main/testutil"
+"testing"
+)
+
+// https://codeforces.com/problemset/problem/2053/B 1200
+// https://codeforces.com/problemset/status/2053/problem/B?friends=on
+func Test_cf2053B(t *testing.T) {
+testCases := [][2]string{
+{
+`5
+2
+1 1
+1 1
+4
+1 3
+1 3
+1 3
+1 3
+6
+3 6
+2 2
+1 2
+1 1
+3 4
+2 2
+7
+3 4
+4 4
+4 4
+1 3
+2 5
+1 4
+2 2
+3
+4 5
+4 4
+5 5`,
+`00
+1111
+100110
+1001111
+011`,
+},
+}
+testutil.AssertEqualStringCase(t, testCases, 0, cf2053B)
+}