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

说明/提示

对于 60%60% 的数据，𝑁,𝑀≤50N,M≤50。

对于 90%90% 的数据，𝑁,𝑀≤300N,M≤300。

对于 100%100% 的数据，1≤𝑁,𝑀≤1031≤N,M≤103，1≤𝐶≤min⁡(𝑁,𝑀)1≤C≤min(N,M)。

基础的二维前缀和，不熟悉的同学可以去了解一下

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<vector>
#include<math.h>
#include<iomanip>
#include<set>
#include<queue>
#include<stack> 
#include<map>
#include<list>
#include <stdlib.h>
#include<deque>
#include <stdlib.h>
#include <time.h>
#include<cstdlib>
using namespace std;
long long n, m,c, a[1005][1005],s[1005][1005],ans=-1;
int x, y;
int main()
{
	cin >> n >> m >> c;
	for (int i = 1; i < n+1; i++)
	{
		for (int j = 1; j < m+1; j++)
		{
			cin >> a[i][j];
			s[i][j] = a[i][j] + s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
		}
	}
	for (int i = c; i <= n; i++)
	{
		for (int j = c; j <= m; j++)
		{
			if ((s[i][j] - s[i][j - c] - s[i - c][j] + s[i - c][j - c]) > ans)
			{
				ans = (s[i][j] - s[i][j - c] - s[i - c][j] + s[i - c][j - c]);
				x = i - c+1, y = j - c+1;
			}
		}
	}
	cout << ans<<endl;
	cout << x << " " << y;
}