CF1401B

Author: EndlessCheng

main/1400-1499/1401B.go

@@ -0,0 +1,33 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1401B(_r io.Reader, _w io.Writer) {
+	in := bufio.NewReader(_r)
+	out := bufio.NewWriter(_w)
+	defer out.Flush()
+	min := func(a, b int) int {
+		if a > b {
+			return b
+		}
+		return a
+	}
+
+	var T, a, b, c, x, y, z int
+	for Fscan(in, &T); T > 0; T-- {
+		Fscan(in, &a, &b, &c, &x, &y, &z)
+		ans := min(c, y) * 2
+		cnt := c - min(c, y) + a
+		if cnt < z {
+			ans -= (z - cnt) * 2
+		}
+		Fprintln(out, ans)
+	}
+}
+
+//func main() { CF1401B(os.Stdin, os.Stdout) }

main/1400-1499/1401B_test.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1401/B
+// https://codeforces.com/problemset/status/1401/problem/B
+func TestCF1401B(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+3
+2 3 2
+3 3 1
+4 0 1
+2 3 0
+0 0 1
+0 0 1
+outputCopy
+4
+2
+0`
+	testutil.AssertEqualCase(t, rawText, 0, CF1401B)
+}