pta 训练 7-2 奥马尔·爱糖果 (20分) 二维前缀和+倒着读入矩阵

奥马尔喜欢吃很多糖果,但不幸的是,大部分的糖果都不健康。所以他的父母找到了一种方法,给每个糖果打分,分数越高,糖果就越健康(分数是一个整数,可以是正的、零的或负的)。一个孩子和他的父母买了一些糖果,他们到了卖糖果的地方,所有的糖果都被存储在一个N*M的二维网格中,每一行有M个糖果。行从上到下从1到N编号,列从左到右从1到M编号,每个单元格包含一个糖果健康值。奇怪的是,这个二维网格中的糖果的健康值总是从上到下、从左到右依次增大,如下图

捕获2.PNG
在这里插入图片描述

奥马尔想买完一个子矩阵中的所有糖果,请你编程序求出他所能买得的最大的糖果健康值。
输入格式:

输入的第一行是单个整数T,测试用例的数量(1≤T≤100)。每个测试用例开始一行包含空格隔开两个整数N M (1≤N,M≤1000)代表糖果网格的尺寸,紧随其后的是N行,每一个包含M个空格隔开整数,代表这一行糖果的健康值。网格中的每个整数将不小于- 2000,也不大于2000。
输出格式:

对于每个测试用例,打印包含单个整数的一行,该整数表示从非空子矩形中可以获得的健康值的最大总和。
输入样例:

在这里给出一组输入。例如:

1
3 3
-4 -2 -1
-3 2 3
1 4 5

输出样例:

在这里给出相应的输出。例如:

14


题意:给定一个有序矩阵(从上到下、从左到右依次增大),求最大子矩阵

  1. 有序,可以发现答案一定是右下角的某个矩阵
  2. 我们可以用二维前缀和求矩阵的和
  3. 倒着读入矩阵会更方便处理
  4. 思考是否有一边读入,一边求右下角前缀和的方法 (mark一下)
	read(Q);
	while(Q--) {
		read(n, m);
		int ans = -INF;
		for(int i=0; i<n; i++) //倒向读入矩阵
			for(int j=0; j<m; j++) {
				int r = n-i, c = m-j;
				read(a[r][c]);
			}
		for(int i=1; i<=n; i++) //正向求二维前缀和, 取最值
			for(int j=1; j<=m; j++) {
				psum[i][j] = psum[i-1][j] + psum[i][j-1] + 
							 a[i][j] - psum[i-1][j-1];
				ans = max(ans, psum[i][j]);
			}
		printf("%d\n", ans);
	}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值