二维前缀和的练习

题目描述

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

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

输入格式

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

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

输出格式

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

原题地址：P2004 领地选择 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<bits/stdc++.h>
using namespace std;
const int N=1005;
int a[N][N],s[N][N];
int main() {
    int n,m,c; cin>>n>>m>>c;        //n是地图的宽，m是地图的长，c是首都的边长
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            cin >> a[i][j];
            s[i][j] = s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];  //二维前缀和sum[][]存在以下递推关系：
                                                                     //sum[i][j] = sum[i-1][j]+sum[i][j-1]-sum[i-1][j1]+a[i][j]
                                                                //根据这个递推关系，用两种for循环可以算出sum[][]。
        }
    int Max = -1<<30, x, y;
    for(int x1=1;x1<=n-c+1;x1++)
        for(int y1=1;y1<=m-c+1;y1++){                 //枚举所有坐标点
            int x2=x1+c-1,y2=y1+c-1;                   //正方形右下角坐标
            int ans = s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1];
            if(ans > Max){
                Max = ans;
                x=x1, y=y1;
            }
        }
    cout<<x<<" "<<y<<"\n";
    return 0;
}