题目大意
平面上给N个点,有M组询问,每组询问是给定一个坐标中心,水平垂直作出两条坐标轴,求第一象限+第三象限点的个数减去第二象限+第四象限点的个数的绝对值。
解题思路
N和M的范围非常大,所以采用离线处理的方法,一次性就把所有的答案计算出来。
考虑下,如果把所有点按照x坐标排序,然后把所有询问按照x坐标排序,那么当我们处理某一个询问的时候,在这个询问的点左边的所有点我们是可以得到的,同时我们也可以用权值线段树或者权值树状数组,利用降维那种方法得到y坐标的分布,只要我们按照x大小顺序把每一个点的y坐标放进线段树。这样的话,我们只需要询问从0到当前询问的y坐标范围内有多少个点,就可以得到第三象限的点,同时第二象限的点也就得出来了。与此同时,我们可以反着推第四象限点数,然后同时得出了第一象限点数,然后答案就得到了。时间复杂度是O(n*log(500000))的,完全可以接受。
参考代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <vector>
using namespace std;
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
const int maxn = (int)5e5 + 10;
const int N = 500000;
int n, m, cs;
struct node{
int x, y;
node(){}
node(int a, int b) : x(a), y(b){}
bool operator < (const node &rhs)const{
return x < rhs.x;
}
};
struct Query{
int x, y, i;
Query(){}
Query(int a, int b, int c) : x(a), y(b), i(c){}
bool operator < (const Query &rhs)const{
return x < rhs.x;
}
};
struct segtree{
int cnt[maxn << 2];
void build(){
memset(cnt, 0, sizeof(cnt));
}
void update(int pos, int l, int r, int rt){
cnt[rt] += 1;
if (l == r) return;
int m = (l + r) >> 1;
if (pos <= m) update(pos, lson);
else update(pos, rson);
}
int query(int ll, int rr, int l, int r, int rt){
if (ll <= l && rr >= r) return cnt[rt];
int m = (l + r) >> 1;
int ret = 0;
if (ll <= m) ret += query(ll, rr, lson);
if (rr > m) ret += query(ll, rr, rson);
return ret;
}
};
vector<node> pt;
vector<Query> qy;
int n1[50010], n2[50010], n3[50010], n4[50010];
segtree t;
void init(){
t.build(); pt.resize(0); qy.resize(0);
memset(n1, 0, sizeof(n1)); memset(n2, 0, sizeof(n2));
memset(n3, 0, sizeof(n3)); memset(n4, 0, sizeof(n4));
}
void solve(){
init();
scanf("%d%d", &n, &m);
for (int i = 0; i < n; i++){
int x, y;
scanf("%d%d", &x, &y);
pt.push_back(node(x, y));
}
sort(pt.begin(), pt.end());
for (int i = 0; i < m; i++){
int x, y;
scanf("%d%d", &x, &y);
qy.push_back(Query(x, y, i));
}
sort(qy.begin(), qy.end());
int pos = 0;
for (int i = 0; i < qy.size(); i++){
while(pt[pos].x < qy[i].x && pos < pt.size()){
t.update(pt[pos].y, 0, N, 1);
pos += 1;
}
int a3 = t.query(0, qy[i].y, 0, N, 1);
n3[qy[i].i] = a3; n2[qy[i].i] = t.cnt[1] - a3;
}
t.build(); reverse(pt.begin(), pt.end()); reverse(qy.begin(), qy.end());
pos = 0;
for (int i = 0; i < qy.size(); i++){
while(pt[pos].x > qy[i].x && pos < pt.size()){
t.update(pt[pos].y, 0, N, 1);
pos += 1;
}
int a4 = t.query(0, qy[i].y, 0, N, 1);
n4[qy[i].i] = a4; n1[qy[i].i] = t.cnt[1] - a4;
}
for (int i = 0; i < m; i++){
int ans = abs(n1[i] + n3[i] - n2[i] - n4[i]);
printf("%d\n", ans);
}
if (cs != 0) puts("");
}
int main(){
scanf("%d", &cs);
while(cs--) solve();
return 0;
}