1. 程式人生 > >BZOJ 1218(二維字首和)

BZOJ 1218(二維字首和)

傳送門:

題面:

1218: [HNOI2003]鐳射炸彈

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 4197  Solved: 1788
[Submit][Status][Discuss]

Description

一種新型的鐳射炸彈,可以摧毀一個邊長為R的正方形內的所有的目標。現在地圖上有n(N<=10000)個目標,用整數Xi,Yi(其值在[0,5000])表示目標在地圖上的位置,每個目標都有一個價值。鐳射炸彈的投放是通過衛星定位的,但其有一個缺點,就是其爆破範圍,即那個邊長為R的正方形的邊必須和x,y軸平行。若目標位於爆破正方形的邊上,該目標將不會被摧毀。 

Input

輸入檔案的第一行為正整數n和正整數R,接下來的n行每行有3個正整數,分別表示xi,yi,vi

Output

輸出檔案僅有一個正整數,表示一顆炸彈最多能炸掉地圖上總價值為多少的目標(結果不會超過32767)。

Sample Input

2 1
0 0 1
1 1 1

Sample Output

1

題目分析:

    維護一下二維字首和,列舉正方形邊長即可。複習一下二維字首和的寫法。不懂可以借鑑一下:傳送門

程式碼:

#include <bits/stdc++.h>
#define maxn 5005
using namespace std;
int sum[maxn][maxn];
int main()
{
    int n,r;
    scanf("%d%d",&n,&r);
    for(int i=0;i<n;i++){
        int x,y,v;
        scanf("%d%d%d",&x,&y,&v);
        sum[x+1][y+1]=v;
    }
    for(int i=1;i<=5001;i++){
        for(int j=1;j<=5001;j++){
            sum[i][j]+=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1];
        }
    }
    int maxx=0;
    for(int i=r;i<=5001;i++){
        for(int j=r;j<=5001;j++){
            maxx=max(maxx,sum[i][j]-sum[i-r][j]-sum[i][j-r]+sum[i-r][j-r]);
        }
    }
    printf("%d\n",maxx);
}