激光炸弹(UPC二维前缀和)

题目描述

一种新型的激光炸弹,可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n(n≤10000)个目标,用整数xi,yi(0≤xi,yi≤5000)表示目标在地图上的位置,每个目标都有一个价值0<vi<100。激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆破范围,即那个边长为R的正方形的边必须和x,y轴平行。若目标位于爆破正方形的边上,该目标将不会被摧毁。

现在你的任务是计算一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入

第一行为正整数n和正整数R,接下来的n行每行有3个正整数,分别表示xi,yi,vi。

输出

仅有一个正整数,表示一颗炸弹最多能炸掉地图上总价值为多少的目标(结果不会超过32767)。

样例输入

扫描二维码关注公众号,回复: 2635395 查看本文章
2 1
0 0 1
1 1 1

样例输出

1

       题目意思比较简单,就是给一个n*n的矩阵,矩阵里面有一些点有权值,在给一个边长为r的正方形,问正方形最多可以包括多少权值。这个题目要求正方形必须与坐标平行,所以我们可以试一下用二维前缀和来维护矩形里面的权值,先处理一下矩阵,然后枚举每个正方形,求出最大值,预处理方程为 f[i][j] = f[i][j]+f[i-1][j]+f[i][j-1]-f[i-1][j-1]; 求ans的方程为ans = max(ans, f[i][j]-f[i-r][j]-f[i][j-r]+f[i-r][j-r])。

AC代码:

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int maxn = 5000 + 50;
typedef long long ll;
int f[maxn][maxn];
int main()
{
    int n, r;
    while(scanf("%d%d", &n, &r) != EOF)
    {
        if(r == 0)
        {
            printf("0\n");
            continue;
        }
        int x, y, v;
        memset(f, 0, sizeof(f));
        for(int i = 0; i < n; ++ i)
        {
            scanf("%d%d%d", &x, &y, &v);
            f[x+1][y+1] = v;
        }
        for(int i = 1; i <= 5001; ++ i)
        {
            for(int j = 1; j <= 5001; ++ j)
            {
                f[i][j] = f[i][j]+f[i-1][j]+f[i][j-1]-f[i-1][j-1];
            }
        }
        int ans = 0;
        for(int i = r; i <= 5001; ++ i)
        {
            for(int j = r; j <= 5001; ++ j)
            {
                ans = max(ans, f[i][j]-f[i-r][j]-f[i][j-r]+f[i-r][j-r]);
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/MALONG11124/article/details/81152185