POJ3416 Crossing

题目链接

题目大意

平面上给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;
}