在一个星光摧残的夜晚,蒜头君一颗一颗的数这天上的星星。
蒜头君给在天上巧妙的画了一个直角坐标系,让所有的星星都分布在第一象。天上有 nn 颗星星,他能知道每一颗星星的坐标和亮度。
现在,蒜头君问自己 q 次,每次他问自己每个矩形区域的星星的亮度和是多少(包含边界上的星星)。
输入格式
第一行输入一个整数 n(1≤n≤50000) 表示星星的数量。
接下里 n 行,每行输入三个整数x,y,w(0≤x,y,w≤2000),表示在坐标(x,y) 有一颗亮度为 w 的星星。注意一个点可能有多个星星。
接下来一行输入一个整数 q(1≤q≤50000),表示查询的次数。
接下来q 行,每行输入四个整数x1,y1,x2,y2,其中(x1,y1) 表示查询的矩形的左下角的坐标,(x2,y2) 表示查询的矩形的右上角的坐标,0≤x1≤x2≤2000,0≤y1≤y2≤2000。
输出格式
对于每一次查询,输出一行一个整数,表示查询的矩形区域内的星星的亮度总和。
样例输入
5 5 0 6 7 9 7 8 6 13 9 7 1 3 0 19 4 0 8 7 9 0 0 7 10 2 7 10 9 5 4 7 5
样例输出
7 32 8 0
思路:
如图,红色区域 = 大矩形 - 矩形1 - 矩形2 + 矩形3 , 需要注意边界问题(虚线)dp[i][j] 代表(0, 0)~(i,j)的所有亮度。
#include<bits/stdc++.h>
using namespace std;
const int N = 2000;
int mp[N+3][N+3], dp[N+3][N+3];
int main()
{
int n, i, j, m, a, b, w;
cin>>n;
while(n--)
{
scanf("%d%d%d", &a, &b, &w);
mp[a+1][b+1] += w;//坐标+1防止下标越界
}
for(i = 1; i<=N; i++)
for(j = 1; j<=N; j++)
dp[i][j] = dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1] + mp[i][j];
cin>>m;
while(m--)
{
int x1,y1, x2,y2;
scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
x1++, y1++, x2++, y2++;
printf("%d\n", dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1]);
}
return 0;
}