牛客 14505 轟炸區最優選取 (二維字首和)

題目連結：轟炸區最優選取

題目描述

現在給出一個正方形地圖，其邊長為n，地圖上有的地方是空的，有的地方會有敵人。

我們現在有一次轟炸敵人的機會，轟炸敵人的區域是一個k*k的正方形區域，現在需要你解決的問題就是計算最多轟炸的敵人數量是多少。

輸入描述:

本題包含多組資料，每組資料第一行輸入兩個數n，k。

接下來n行，每行n個數字，表示這個點上的敵人數量。

資料範圍：

1<=n<=50

1<=k<=n

每個點上的敵人數量不超過100個（0<=a[i][j]<=100）。

輸出描述:

每組資料輸出包含一行，表示計算的結果。

示例1

輸入

4 2
1 1 0 0
1 1 0 0
0 0 2 2
0 0 2 2
           

輸出

8
           

說明

樣例中，顯然轟炸右下角那個部分能夠擊敗最多的敵人

題解

二維字首和

二維字首和的模闆題。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 110;
int a[maxn][maxn];
int sum[maxn][maxn];

int main() {
    int n, k;
    while(cin >> n >> k) {
        for(int i = 1; i <= n; ++i) {
            for(int j = 1; j <= n; ++j) {
                scanf("%d", &a[i][j]);
                sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + a[i][j];
            }
        }
        int ans = 0;
        for(int i = k; i <= n; ++i) {
            for(int j = k; j <= n; ++j) {
                int s = sum[i][j] - sum[i - k][j] - sum[i][j - k] + sum[i - k][j - k];
                ans = max(ans, s);
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}