轰炸区域最优选取(二维前缀和)

来源:牛客网

现在给出一个正方形地图,其边长为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;
int a[55][55], mapp[55][55];
int main() {
    
    
	int n, k, x;
	while(cin>>n>>k) {
    
    
		for(int i=1; i<=n; i++) {
    
    
			for(int j=1; j<=n; j++) {
    
    
				cin>>a[i][j];
				mapp[i][j]=mapp[i][j-1]+mapp[i-1][j]-mapp[i-1][j-1]+a[i][j];
			}
		}
		int maxm=0;
		for(int i=k; i<=n; i++) {
    
    
			for(int j=k; j<=n; j++) {
    
    
				maxm=max(maxm,mapp[i][j]-mapp[i][j-k]-mapp[i-k][j]+mapp[i-k][j-k]);
			}
		}
		cout<<maxm<<endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45465598/article/details/110701119