洛谷 P2004 领地选择（二维前缀和）

 题目链接：

领地选择 - 洛谷https://www.luogu.com.cn/problem/P2004

 思路：

二维前缀和，sum[ i ][ j ]记录从以(1,1)和(i, j)为左上角，右下角组成的矩形中的value之和，由此容易计算出结果（计算方法见代码）

代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 1e3+5;
int n, m, c;
int sum[maxn][maxn]; //记录前缀和
int mp[maxn][maxn]; //记录当地作为首都的总价值

signed main(){
    ios::sync_with_stdio(false);
    cin >> n >> m >> c;

    //输入value并计算前缀和，计算前缀和先整合行，再整合列
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            cin >> sum[i][j];
            sum[i][j] += sum[i][j-1];
        }
    }
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            sum[i][j] += sum[i-1][j];
        }
    }

    int maxx, maxy;
    int maxVal = -(1<<30);
    for(int i=1; i<=n-c+1; i++){
        for(int j=1; j<=m-c+1; j++){
            mp[i][j] = sum[i+c-1][j+c-1]-sum[i+c-1][j-1]-sum[i-1][j+c-1]+sum[i-1][j-1];
            if(mp[i][j]>maxVal){
                maxVal = mp[i][j];
                maxx = i, maxy = j;
            }
        }
    }
    cout << maxx << ' ' << maxy<<'\n';
}