Merge remote-tracking branch 'origin/master'

Author: EndlessCheng

copypasta/bits.go

@@ -421,10 +421,14 @@ func bitsCollection() {
 	}
 
 	//（接上）考虑乘法
-	// 给一数组 a，元素均为正整数，求区间和等于区间积的区间个数
-	// 核心思路是对于每个区间右端点，其对应的区间积不会超过 sum(a)，且由于乘积至少要乘 2 才会变化，所以区间右端点对应的区间积至多有 O(log(sum(a))) 个不同的值
-	// 而每个前缀和都是不同的，所以区间右端点对应的答案个数也至多有 O(log(sum(a))) 个
-	// 因此总的答案个数至多为 O(nlog(sum(a)))
+	// 问题：给一数组 a，元素均为正整数，求区间和等于区间积的区间个数
+	// 我们来考虑对每个区间右端点，有多少个合法的区间左端点
+	// 核心思路是，对于每个满足题目要求的区间，其区间积不会超过 sum(a)
+	// 由于乘积至少要乘 2 才会变化，所以对于一个固定的区间右端点，不同的区间积至多有 O(log(sum(a))) 个
+	// 同时由于元素均为正数，所以对一个固定的区间右端点，区间左端点也至多有 O(log(sum(a))) 个
+	// 据此我们只需要在加入一个新的数后，去重并去掉区间积超过 sum(a) 的区间，就可以暴力做出此题
+	// 注：根据以上推导过程，我们还可以得出总的答案个数至多为 O(nlog(sum(a)))
+	// https://www.dotcpp.com/oj/problem2622.html
 	countSumEqMul := func(a []int) (ans int) {
 		tot := 0
 		for _, v := range a {

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)
+}