2018蓝桥模拟赛·天上的星星 暴力|二维树状数组

在一个星光摧残的夜晚，蒜头君一颗一颗的数这天上的星星。

蒜头君给在天上巧妙的画了一个直角坐标系，让所有的星星都分布在第一象。天上有 $n$ 颗星星，他能知道每一颗星星的坐标和亮度。

现在，蒜头君问自己 $q$ 次，每次他问自己每个矩形区域的星星的亮度和是多少（包含边界上的星星）。

输入格式

第一行输入一个整数 $n(1\le n\le 50000)$ 表示星星的数量。

接下里 $n$ 行，每行输入三个整数 $x,y,w(0\le x,y,w\le 2000)$，表示在坐标 $(x,y)$ 有一颗亮度为 $w$ 的星星。注意一个点可能有多个星星。

接下来一行输入一个整数 $q(1\le q\le 50000)$，表示查询的次数。

接下来 $q$ 行，每行输入四个整数 x_1, y_1, x_2, y_2x1​,y1​,x2​,y2​，其中 (x_1, y_1)(x1​,y1​) 表示查询的矩形的左下角的坐标，(x_2, y_2)(x2​,y2​) 表示查询的矩形的右上角的坐标，0 \le x_1 \le x_2 \le 20000≤x1​≤x2​≤2000，0 \le y_1 \le y_2 \le 20000≤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

给出第一象限下的坐标并给出这个坐标下的值 给我们多个矩形的范围 求这个

虽然暴力能过

但是树状数组 也可以~并且更快~

复杂度O(q*log(x)*log(y))

x,y<=2000

注意边界要算进去 而且数据不能存在0,0上

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll tre[2010][2010];
void add(int x,int y,int w)
{
	for(int i=x;i<=2001;i+=i&(-i)){
		for(int j=y;j<=2001;j+=j&(-j)){
			tre[i][j]+=w;
	}
}
ll query(int x,int y){
	ll sum=0;
	for(int i=x;i>0;i-=i&(-i)){
		for(int j=y;j>0;j-=j&(-j)){
			sum+=tre[i][j];			
	}
	return sum; 
}
int main()
{
	int n;
	ios::sync_with_stdio(0);
	cin>>n;
	while(n--){
		int x,y,w;
		cin>>x>>y>>w;
		add(x+1,y+1,w);
	}
	int q;
	cin>>q;
	while(q--){
		int lx,ly,rx,ry;
		cin>>lx>>ly>>rx>>ry;
		
		ll ans = query(rx+1,ry+1)-query(rx+1,ly)-query(lx,ry+1)+query(lx,ly); //注意边要算在矩形内 
		cout<<ans<<endl;			
	}
	return 0;
}