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

题意：给一个n个二元组：<a,b>，给m个三元组：<c,d,e>。如果b=e那么可以合并二元组，三元组，得到一个新的三元组：<a,c,d>。

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

分析：学习的这里的做法

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

扫一遍所有的三元组：二位线段树维护<c,d>的值，查找当前线段树中有没有元素<x,y>满足x≥c,y≥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;}