洛谷[P2004 领地选择] {前缀和与差分} 奋斗的珂珂~

洛谷[P2004 领地选择]

题目描述

作为在虚拟世界里统帅千军万马的领袖，小 Z 认为天时、地利、人和三者是缺一不可的，所以，谨慎地选择首都的位置对于小 Z 来说是非常重要的。

首都被认为是一个占地 C×C 的正方形。小 Z 希望你寻找到一个合适的位置，使得首都所占领的位置的土地价值和最高。

输入格式

第一行三个整数N,M,C，表示地图的宽和长以及首都的边长。

接下来 N 行每行 M 个整数，表示了地图上每个地块的价值。价值可能为负数。

输出格式

一行两个整数 X,Y，表示首都左上角的坐标。

输入输出样例

输入
3 4 2
1 2 3 1
-1 9 0 2
2 0 1 1
输出
1 2

说明/提示

对于 60% 的数据，N,M≤50。

对于 90%90% 的数据，N,M≤300。

对于 100%100% 的数据，1≤N,M≤$10^3$ ,1≤C≤min(N,M)。

解题思路

     首先，使用**b[i][j]来表示矩阵i*j覆盖下的价值和。b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j]，求解b[i][j]如图：
因为b[i-1][j]+b[i][j-1]多加了b[i-1][j-1],所以要减去，而且还要加上当前位置的值a[i][j]。

然后，求解矩阵（正方形）覆盖下的价值和，等于b[i][j]-b[i-C][j]-b[i][j-C]+b[i-C][j-C]（C代表正方形的边长）。
表达式解释由图所示：（C=2）

因为-b[i-C][j]-b[i][j-C]相当于减去两次的b[i-C][j-C]，所以需要加回来一次。

完整代码

#include<bits/stdc++.h>
using namespace std;
int a[3000][3000],b[3000][3000];

int main()
{
	int N,M,C;
	scanf("%d %d %d",&N,&M,&C);
	for(int i=1;i<=N;i++)
	  for(int j=1;j<=M;j++)
	  {
	  	scanf("%d",&a[i][j]);
	  	b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j];//b[i][j]表示i*j的矩阵覆盖下的总价值 
	  }
    int  ans=-1e9;//题目中说明价值可能为负数 
    int p,q;
    for(int i=C;i<=N;i++)
	  for(int j=C;j<=M;j++)
	  {
	  		if(ans<b[i][j]-b[i-C][j]-b[i][j-C]+b[i-C][j-C]) //如果C*C矩阵的覆盖下的值大，则进行处理 
	  		{
	  		  p=i-C+1,q=j-C+1;//记录当前状态下最大价值的左上角坐标	
			  ans=b[i][j]-b[i-C][j]-b[i][j-C]+b[i-C][j-C];
		    }
	  }
	  printf("%d %d",p,q);
	  return 0;
 }