P2004 领地选择(二维前缀)

P2004 领地选择

原题戳这里嗷

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

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

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

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

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

输入输出样例
输入 #1
3 4 2
1 2 3 1
-1 9 0 2
2 0 1 1
输出 #1
1 2

题解

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010;
int n, m, r, g[N][N], res = -0x7fffffff, pos1, pos2;

int main(){
    cin >> n >> m >> r;
    //前缀和
    for (int i = 1 ; i <= n ; i ++ ){
        for (int j = 1 ; j <= m ; j ++ ){
            cin >> g[i][j];
            g[i][j] += g[i - 1][j] + g[i][j - 1] - g[i - 1][j - 1];
        }
    }
    //遍历，注意起始位置为r
    for (int i = r; i <= n ; i ++ ){
        for (int j = r ; j <= m ; j ++){
            if (res < g[i][j] - g[i - r][j] - g[i][j - r] + g[i - r][j - r]){
                res = g[i][j] - g[i - r][j] - g[i][j - r] + g[i - r][j - r];
                pos1 = i - r + 1;
                pos2 = j - r + 1;
            }
        }
    }
    
    cout << pos1 << ' ' << pos2;
    
    
    return 0;
}

• 二维前缀和：画矩形

和某P2280激光炸弹思路一摸一样啊，二维前缀模板题