Merge remote-tracking branch 'origin/master'

Author: EndlessCheng

copypasta/template/interactive_problem/main.go

@@ -9,7 +9,7 @@ import (
 // github.com/EndlessCheng/codeforces-go
 type (
 	initData struct{ n int }
-	request  struct{ i int }
+	request  struct{ q []int }
 	response struct{ v int }
 	answer   struct{ ans []int }
 )
@@ -30,9 +30,14 @@ func (io io) readInitData() (d initData) {
 	return
 }
 
-func (io io) query(req request) (resp response) {
-	Fprintln(io.out, "?", req.i)
+func (io io) query(q request) (resp response) {
+	Fprint(io.out, "?")
+	for _, v := range q.q {
+		Fprint(io.out, " ", v)
+	}
+	Fprintln(io.out)
 	io.out.Flush()
+
 	Fscan(io.in, &resp.v)
 	if resp.v < 0 {
 		panic(-1)
@@ -41,9 +46,10 @@ func (io io) query(req request) (resp response) {
 }
 
 func (io io) printAnswer(a answer) {
-	Fprint(io.out, "! ")
+	Fprint(io.out, "!")
+	//Fprint(io.out, " ", len(a.ans))
 	for _, v := range a.ans {
-		Fprint(io.out, v, " ")
+		Fprint(io.out, " ", v)
 	}
 	Fprintln(io.out)
 	io.out.Flush()
@@ -56,7 +62,7 @@ func (io io) printAnswer(a answer) {
 }
 
 func doInteraction(it interaction) {
-	q := func(i int) int { return it.query(request{i}).v }
+	q := func(q ...int) int { return it.query(request{q}).v }
 	dt := it.readInitData()
 	n := dt.n
 	ans := make([]int, n) //

copypasta/template/interactive_problem/main_test.go

@@ -18,18 +18,18 @@ var failedCount int
 type mockIO struct {
 	initData
 	answer
-	innerData []int
+	hiddenData []int //
 
 	_t         *testing.T
 	caseNum    int
 	queryLimit int
 	queryCnt   int
 }
 
-func (io *mockIO) String() (s string) {
-	s = Sprintf("%v", io.innerData)
-	//s = strings.Join(io.innerData, "\n")
-	return
+func (io *mockIO) String() string {
+	hStr := Sprintf("%v", io.hiddenData)
+	//hStr := strings.Join(io.hiddenData, "\n")
+	return Sprintf("%+v\n%s", io.initData, hStr)
 }
 
 // Mock initData
@@ -40,9 +40,9 @@ func (io *mockIO) readInitData() (d initData) {
 // Check answer
 func (io *mockIO) printAnswer(actualAns answer) {
 	expectedAns := io.answer
-	if !assert.EqualValues(io._t, expectedAns, actualAns, "Wrong Answer %d\nInner Data:\n%v", io.caseNum, io) {
+	if !assert.EqualValues(io._t, expectedAns, actualAns, "Wrong Answer %d\nCase Data:\n%v", io.caseNum, io) {
 		if failedCount++; failedCount > failedCountLimit {
-			io._t.Fatal("too many wrong answers, terminated")
+			io._t.Fatal("too many failed cases, terminated")
 		}
 	}
 
@@ -51,24 +51,24 @@ func (io *mockIO) printAnswer(actualAns answer) {
 
 		return true
 	}
-	if !assert.Truef(io._t, ansChecker(), "Wrong Answer %d\nMy Answer:\n%v\nInner Data:\n%v", io.caseNum, actualAns, io) {
+	if !assert.Truef(io._t, ansChecker(), "Wrong Answer %d\nMy Answer:\n%v\nCase Data:\n%v", io.caseNum, actualAns, io) {
 		if failedCount++; failedCount > failedCountLimit {
-			io._t.Fatal("too many wrong answers, terminated")
+			io._t.Fatal("too many failed cases, terminated")
 		}
 	}
 }
 
 // Mock query
-func (io *mockIO) query(req request) (resp response) {
+func (io *mockIO) query(q request) (resp response) {
 	if io.caseNum == debugCaseNum {
-		Print("Query ", req, " ")
+		Print("Query ", q, " => ")
 		defer func() { Println(resp) }()
 	}
-	if io.queryCnt++; io.queryCnt > io.queryLimit {
-		io._t.Fatalf("Query Limit Exceeded %d\nInner Data:\n%v", io.caseNum, io)
-	}
 
+	io.queryCnt++
+	//if io.queryCnt > io.queryLimit { io._t.Fatalf("Query Limit Exceeded %d\nCase Data:\n%v", io.caseNum, io) }
 
+	// calc q ...
 
 	return
 }
@@ -81,17 +81,27 @@ func Test_doInteraction(_t *testing.T) {
 		}
 		io := &mockIO{_t: _t, caseNum: tc}
 
+		// gen random data ...
 		rg = testutil.NewRandGenerator()
 		n := rg.Int(2, 4)
 		a := rg.IntSlice(n, 1, n)
 
 		io.n = n
 		io.ans = a
-		io.innerData = a
+		io.hiddenData = a
 
+		// set limit ...
 		io.queryLimit = n + 30
 
 		doInteraction(io)
+
+		if io.queryCnt > io.queryLimit {
+			io._t.Errorf("Query Limit Exceeded %d\n%d > %d\nCase Data:\n%v", io.caseNum, io.queryCnt, io.queryLimit, io)
+			if failedCount++; failedCount > failedCountLimit {
+				io._t.Fatal("too many failed cases, terminated")
+			}
+		}
+
 		if tc == checkTC {
 			_t.Logf("%d cases checked.", tc)
 			checkTC <<= 1

leetcode/weekly/272/a/a.go

@@ -0,0 +1,17 @@
+package main
+
+// 遍历，找第一个回文串
+
+// github.com/EndlessCheng/codeforces-go
+func firstPalindrome(words []string) string {
+next:
+	for _, w := range words {
+		for i, n := 0, len(w); i < n/2; i++ {
+			if w[i] != w[n-1-i] {
+				continue next
+			}
+		}
+		return w
+	}
+	return ""
+}

leetcode/weekly/272/a/a_test.go

@@ -0,0 +1,27 @@
+package main
+
+// Go 模拟
+
+// github.com/EndlessCheng/codeforces-go
+func addSpaces(s string, spaces []int) string {
+	spaces = append(spaces, len(s)) // 小技巧：把 s 长度加到数组末尾，这样可以在循环内处理最后一段
+	ans := []byte(s[:spaces[0]])
+	for i := 1; i < len(spaces); i++ {
+		ans = append(ans, ' ')
+		ans = append(ans, s[spaces[i-1]:spaces[i]]...)
+	}
+	return string(ans)
+}
+
+// github.com/EndlessCheng/codeforces-go
+func addSpaces2(s string, spaces []int) string {
+	ans := make([]byte, 0, len(s)+len(spaces))
+	ans = append(ans, s[:spaces[0]]...)
+	for i := 1; i < len(spaces); i++ {
+		ans = append(ans, ' ')
+		ans = append(ans, s[spaces[i-1]:spaces[i]]...)
+	}
+	ans = append(ans, ' ')
+	ans = append(ans, s[spaces[len(spaces)-1]:]...)
+	return string(ans)
+}

leetcode/weekly/272/b/b.go

@@ -0,0 +1,29 @@
+package main
+
+/* 分组循环
+
+将 $\textit{prices}$ 按照平滑下降的定义分成若干组。例如 $[3,2,1,4]$ 分为 $[3,2,1]$ 和 $[4]$ 两组。
+
+对于每一组的所有非空子数组，都是平滑下降的。设该组长度为 $m$，则该组的非空子数组个数为
+
+$$
+C_{m+1}^2 = \dfrac{m(m+1)}{2}
+$$
+
+累加每组的非空子区间个数即为答案。
+
+- 时间复杂度：$O(n)$，其中 $n$ 是数组 $\textit{prices}$ 的长度。注意下面代码内外层循环共用同一个变量 $i$，时间复杂度就是 `i++` 的执行次数，即 $O(n)$。
+- 空间复杂度：$O(1)$，我们只需要常数的空间保存若干变量。
+
+*/
+
+// github.com/EndlessCheng/codeforces-go
+func getDescentPeriods(prices []int) (ans int64) {
+	for i, n := 0, len(prices); i < n; {
+		i0 := i
+		for i++; i < n && prices[i] == prices[i-1]-1; i++ {
+		}
+		ans += int64(i-i0) * int64(i-i0+1) / 2
+	}
+	return
+}

leetcode/weekly/272/b/b_test.go

@@ -0,0 +1,42 @@
+package main
+
+import "sort"
+
+/* 最长上升子序列
+
+将 $\textit{arr}$ 每隔 $k$ 个数取一个元素，分为若干组，例如当 $k=3$ 时，分为如下三组：
+
+- $arr[0],arr[3],arr[6],\cdots$
+- $arr[1],arr[4],arr[7],\cdots$
+- $arr[2],arr[5],arr[8],\cdots$
+
+根据题意，组与组之间是互不影响的，我们需要让每一组内的元素单调不降。
+
+题目要求需要修改的最少元素个数，这可以通过求不需要修改的最多元素个数来求出，也就是该组元素的最长不降子序列。（不了解的同学可以看看第 300 题）
+
+需要注意的是：
+
+- 求最长严格递增子序列需要二分找到大于或等于当前元素的元素位置（即 C++ 中的 `lower_bound`）；
+- 求最长非降子序列需要二分找到大于当前元素的元素位置（即 C++ 中的 `upper_bound`）。
+
+累加所有组的最长不降子序列的长度即为最多可以保留的元素个数，用 $\textit{arr}$ 的长度减去该个数即为答案。
+
+*/
+
+// github.com/EndlessCheng/codeforces-go
+func kIncreasing(arr []int, k int) int {
+	save := 0
+	for i, n := 0, len(arr); i < k && i < n; i++ {
+		f := []int{}
+		for j := i; j < n; j += k {
+			v := arr[j]
+			if p := sort.SearchInts(f, v+1); p < len(f) {
+				f[p] = v
+			} else {
+				f = append(f, v)
+			}
+		}
+		save += len(f)
+	}
+	return len(arr) - save
+}