传送门:点击打开链接
题意:有一个矩形区域被划分为N行M列的网格,每个格子里有一定数量的资源并记录在矩阵val中,坐标(x,y)位置上资源量为val[x][y],其val中每个元素的值为0~9的整数。如果你在某个网格(a,b)上造一座保护塔,那么你可以占领K个网格中的资源,这K个格子分别是(a+dx[1],b+dy[1]),(a+dx[2],b+dy[2]),...,(a+dx[K],b+dy[K]),注意(a,b)这格本身可能未必会被占领。现在你能建造不超过2个塔,问最多能占领多少资源?一个网格被多个塔占领时其资源只计算一次。另外如果计算的位置(a+dx[i],b+dy[i])在网格外,则不贡献任何资源。
思路:如果直接去枚举,肯定是不行的。
我们先考虑去预处理出一个塔能得到的最多资源。
之后,我们去枚举第一个塔的位置,那么第二个塔的位置,有2种可能。
一种是与第一座塔有冲突,一种是与第一座塔没有冲突。
我们可以去枚举有冲突的情况,因为k<=10,我们直接枚举是哪两个地方的资源冲突,复杂度O(k^2)
对于有冲突的塔,我们算出冲突的资源总和,之后要减掉重复的
对于没有冲突的塔,在我们算出哪些冲突以后,剩下的就是没有冲突的了
我们直接用RMQ预处理,然后区间O(1)查询最大
最后总复杂度O(n*m*log(n*m))
#include <map>
#include <set>
#include <cmath>
#include <ctime>
#include <stack>
#include <queue>
#include <cstdio>
#include <cctype>
#include <bitset>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
#define fuck(x) cout<<"["<<x<<"]";
#define FIN freopen("input.txt","r",stdin);
#define FOUT freopen("output.txt","w+",stdout);
//#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int MX = 1e2 + 5;
const int MS = 1e4 + 5;
const int INF = 0x3f3f3f3f;
int n, m, k;
char W[MX][MX];
int A[MS], MAX[MS][15], vis[MX][MX];
int dx[15], dy[15], dq[MS];
void RMQ_init(int n) {
for(int i = 0; i < n + 1; i++) {
MAX[i][0] = A[i];
}
for(int j = 1; (1 << j) <= n + 1; j++) {
for(int i = 0; i + (1 << j) - 1 < n + 1; i++) {
MAX[i][j] = max(MAX[i][j - 1], MAX[i + (1 << (j - 1))][j - 1]);
}
}
}
int RMQ_max(int L, int R) {
int k = 0;
while((1 << (k + 1)) <= R - L + 1) k++;
return max(MAX[L][k], MAX[R - (1 << k) + 1][k]);
}
bool umax(int &a, int b) {
if(b > a) {
a = b; return true;
}
return false;
}
inline int ID(int x, int y) {
return (x - 1) * m + y;
}
int solve() {
int ret = 0;
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= m; j++) {
int Max = 0, id = ID(i, j), dsz = 0;
for(int q = 1; q <= k; q++) {
int nx = i + dx[q], ny = j + dy[q];
if(nx < 1 || nx > n || ny < 1 || ny > m) continue;
vis[nx][ny] = id;
for(int w = 1; w <= k; w++) {
int nnx = nx - dx[w], nny = ny - dy[w];
if(nnx < 1 || nnx > n || nny < 1 || nny > m) continue;
dq[++dsz] = ID(nnx, nny);
}
}
dq[++dsz] = 0;
dq[++dsz] = m * n + 1;
sort(dq + 1, dq + 1 + dsz);
dsz = unique(dq + 1, dq + 1 + dsz) - dq - 1;
for(int q = 1; q <= dsz - 1; q++) {
if(dq[q] + 1 <= dq[q + 1] - 1) umax(Max, RMQ_max(dq[q] + 1, dq[q + 1] - 1));
}
for(int q = 2; q <= dsz - 1; q++) {
int x = (dq[q] - 1) / m + 1, y = (dq[q] - 1) % m + 1, sum = A[ID(x, y)];
for(int w = 1; w <= k; w++) {
int nx = x + dx[w], ny = y + dy[w];
if(nx < 1 || nx > n || ny < 1 || ny > m) continue;
if(vis[nx][ny] == id) sum -= W[nx][ny] - '0';
}
umax(Max, sum);
}
umax(ret, Max + A[id]);
}
}
return ret;
}
int main() {
//FIN;
int T; scanf("%d", &T);
while(T--) {
scanf("%d%d%d", &n, &m, &k);
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= m; j++) {
vis[i][j] = A[ID(i, j)] = 0;
}
}
for(int i = 1; i <= n; i++) scanf("%s", W[i] + 1);
for(int i = 1; i <= k; i++) {
scanf("%d%d", &dx[i], &dy[i]);
}
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= m; j++) {
int id = ID(i, j);
for(int q = 1; q <= k; q++) {
int nx = i + dx[q], ny = j + dy[q];
if(nx < 1 || nx > n || ny < 1 || ny > m) continue;
A[id] += W[nx][ny] - '0';
}
}
}
RMQ_init(n * m);
printf("%d\n", solve());
}
return 0;
}