CF1158A

Author: EndlessCheng

main/1100-1199/1158A.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1158A(_r io.Reader, out io.Writer) {
+	in := bufio.NewReader(_r)
+	var n, m uint
+	var v, ans, fi, se uint64
+	for Fscan(in, &n, &m); n > 0; n-- {
+		Fscan(in, &v)
+		ans += v
+		if v > fi {
+			fi, se = v, fi
+		} else if v > se {
+			se = v
+		}
+	}
+	ans = (ans - fi) * uint64(m)
+	min := uint64(1e9)
+	for ; m > 0; m-- {
+		Fscan(in, &v)
+		ans += v
+		if v < min {
+			min = v
+		}
+	}
+	if fi > min {
+		Fprint(out, -1)
+	} else {
+		if fi < min {
+			ans += fi - se
+		}
+		Fprint(out, ans)
+	}
+}
+
+//func main() { CF1158A(os.Stdin, os.Stdout) }

main/1100-1199/1158A_test.go

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