HDU 5517 Triple （2015沈阳站I题&&二维树状数组）

题意：给一个$n$个二元组：$<a,b>$，给$m$个三元组：$<c,d,e>$。如果$b=e$那么可以合并二元组，三元组，得到一个新的三元组： <a,c,d> <script type="math/tex" id="MathJax-Element-6193"> </script>。

设所有的二元组，三元组组合后得到的新的三元组的集合是$C$。
求$C$集合中满足条件的三元组的个数。条件：设这个三元组为$u$，在这个集合中找不到任何一个元素，使得改元素中每一个数都大于等于$u$中对应的成员。

分析：学习的这里的做法

首先，对于二元组，相同的$b$取最大的$a$，并计算此时$a$的个数。对于三元组，如果$e$存在上一个二元组中，那么记录下新的三元组 <a,c,d> <script type="math/tex" id="MathJax-Element-6202"> </script>。重新组合后，进行新的三元组的排序，按照 $a$排序，$a$相同的，按照 $c$排序，$c$相同的按照 $d$排序，都是从大到小排，合并相同的三元组。

扫一遍所有的三元组：二位线段树维护$<c,d>$的值，查找当前线段树中有没有元素 <x,y> <script type="math/tex" id="MathJax-Element-6209"> </script>满足 $x\ge{c},y\ge{c}$，没有就更新答案，最终把答案插进去线段树。

代码：

#include <bits/stdc++.h>
#define LL long long
#define FOR(i,x,y)  for(int i = x;i < y;++ i)
#define IFOR(i,x,y) for(int i = x;i > y;-- i)

using namespace std;

typedef vector <int> VT;

const int maxn = 100010;
const int maxm = 1010;

int b[maxn],cnt_b[maxn];

struct Point{
    int a,c,d;
    int cnt;
    Point() {}
    Point(int x,int y,int z,int k):   a(x),c(y),d(z),cnt(k) {}
    bool operator < (const Point& rhs) const{
        if(a == rhs.a){
            if(c == rhs.c)  return d > rhs.d;
            return c > rhs.c;
        }
        return a > rhs.a;
    }
    bool operator == (const Point& rhs) const{
        if(a == rhs.a && c == rhs.c && d == rhs.d)  return true;
        return false;
    }
}point[maxn],pp[maxn];

int n,m;

void init(){
    scanf("%d%d",&n,&m);
    FOR(i,0,maxn)   b[i] = cnt_b[i] = 0;
    int x,y,z;
    FOR(i,0,n){
        scanf("%d%d",&x,&y);
        if(b[y] == x)   cnt_b[y] ++;
        else if(b[y] < x)   {b[y] = x; cnt_b[y] = 1;}
    }
    n = 0;
    FOR(i,0,m){
        scanf("%d%d%d",&x,&y,&z);
        if(b[z]){
            pp[++n] = Point(b[z],x,y,cnt_b[z]);
        }
    }
    sort(pp+1,pp+n+1);
    int len = 0;
    point[++len] = pp[1];
    FOR(i,2,n+1){
        if(point[len] == pp[i]){
            point[len].cnt += pp[i].cnt;
        }
        else{
            point[++len] = pp[i];
        }
    }
    n = len;
}

int tree[maxm][maxm];

int lowbit(int x)   {return x & (-x);}

void Modify(int x,int y,int val){
    int rt_x = x,rt_y = y;
    for(rt_x = x;rt_x < maxm;rt_x += lowbit(rt_x)){
        for(rt_y = y;rt_y < maxm;rt_y += lowbit(rt_y)){
            tree[rt_x][rt_y] += val;
        }
    }
}

int query(int x,int y){
    int res = 0;
    int rt_x,rt_y;
    for(rt_x = x;rt_x;rt_x -= lowbit(rt_x)){
        for(rt_y = y;rt_y ;rt_y -= lowbit(rt_y)){
            res += tree[rt_x][rt_y];
        }
    }
    return res;
}

void work(){
    FOR(i,0,maxm)
        FOR(j,0,maxm)   tree[i][j] = 0;
    int ans = 0;
    FOR(i,1,n+1){
        if(!query(1001-point[i].c,1001-point[i].d)){
            ans += point[i].cnt;
        }
        Modify(1001-point[i].c,1001-point[i].d,1);
    }
    printf("%d\n",ans);
}

int main()
{
    //freopen("test.in","r",stdin);
    int T,tCase = 0;  scanf("%d",&T);
    while(T--){
        printf("Case #%d: ",++tCase);
        init();
        work();
    }
    return 0;
}