输入一个n行m列的整数矩阵,再输入q个询问,每个询问包含四个整数x1, y1, x2, y2,表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。
输入格式
第一行包含三个整数n,m,q。
接下来n行,每行包含m个整数,表示整数矩阵。
接下来q行,每行包含四个整数x1, y1, x2, y2,表示一组询问。
输出格式
共q行,每行输出一个询问的结果。
数据范围
1≤n,m≤1000,
1≤q≤200000,
1≤x1≤x2≤n,
1≤y1≤y2≤m,
−1000≤矩阵内元素的值≤1000
输入样例:
3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4
输出样例:
17
27
21
Go 代码 TLE
package main
import "fmt"
func main() {
const N = 1010
var n, m, q int
var s = [N][N]int{}
var a = [N][N]int{}
fmt.Scanf("%d%d%d", &n, &m, &q)
for i := 1; i<= n; i++ {
for j := 1; j <= m; j++ {
fmt.Scanf("%d", &a[i][j])
}
}
for i := 1;i <= n; i++ {
for j := 1; j <= m; j++ {
s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j] // 求前缀和
}
}
for q > 0 {
q--
var x1, y1, x2, y2 int
fmt.Scanf("%d%d%d%d", &x1, &y1, &x2, &y2)
fmt.Printf("%d\n", s[x2][y2] - s[x1-1][y2] -s[x2][y1-1] + s[x1-1][y1-1]) // 算子矩阵的和
}
}
Go实现使用bufio.NewScanner
学习参考资料
1. Golang在OJ系统上的坑-输入相关
2. 1. 1.4 bufio — 缓存IO
2. acwa 的题解
3. acwa 的笔记 - 关于Go的各种超时问题
package main
import (
"bufio"
"fmt"
"os"
"strconv"
"strings"
)
func main() {
const N = 1010
var n, m, q int
var a = [N][N]int{}
var s = [N][N]int{}
fmt.Scanf("%d%d%d", &n, &m, &q)
scanner := bufio.NewScanner(os.Stdin) // 实例化 Scanner
for i := 1; i <= n; i++ {
scanner.Scan() // Scan 方法 该方法好比 iterator 中的 Next 方法
line := strings.Split(scanner.Text(), " ")
for j := 1; j <= m; j++ {
a[i][j], _ = strconv.Atoi(line[j-1])
}
}
for i := 1; i <= n; i++ {
for j := 1; j <= m; j++ {
s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j] // 求前缀和
}
}
var x1, y1, x2, y2 int
var ll [4]int
for q > 0 {
q--
scanner.Scan()
temp := strings.Split(scanner.Text(), " ")
for i, s := range temp {
ll[i], _ = strconv.Atoi(s)
}
x1, y1, x2, y2 = ll[0], ll[1], ll[2], ll[3]
fmt.Printf("%d\n", s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]) // 算子矩阵的和
}
}
总结: 使用 bufio.NewScanner 读取输入,注意以下
bufio.NewScanner的底层buf数组有个最大缓存限制的,是64K,也就是说按照标准的Scan,一行最多64K的数据大小,题目里如果超过这个范围,比如一行20万数据,每个数据范围还是int32内的,那这就至少800K的大小了,还没算上中间的空格,所以默认的buf空间肯定是不够的,这时候需要调用Buffer方法,手动给Scanner分配一个满足题目空间的buf数组,其余照常。使用方法如下:
sc := bufio.NewScanner(os.Stdin) // 实例化 NewScanner
bs := make([]byte, 2000*1024) // 缓存
sc.Buffer(bs, len(bs))
sc.Scan() // Scan 方法 该方法好比 iterator 中的 Next 方法,默认ScanLines 返回一行文本,不包括行尾的换行符
fmt.Println(sc.Text()) // 该方法应该在 Scan 调用后调用.Text 返回的是 string