hdu 5517 Triple(二维树状数组)

题目链接：hdu 5517 Triple

题意：

有n个两元组A，有m个三元组B，然后set C有一个计算方式。

现在让你找set TOP的size。

题解：

理解题意后，显然对于每个b的分组，只有最大的a才有贡献，

然后就可以发现set B中每个元素按照e分组后，只会对应一个a，所以最多有1e5个三元组可能有贡献。

然后将这个三元组排一下序，用二维树状数组搞搞就行了。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;
typedef long long ll;

const int N=1e5+7;
int t,n,m,tot,cas,A[N],C[N];
int c[1007][1007];

int sum(int x,int y)
{
    int ret = 0;
    for(int i = x;i > 0;i -= i&-i)
        for(int j = y;j > 0;j -= j&-j)
            ret += c[i][j];
    return ret;
}
void add(int x,int y,int val)
{
    for(int i = x;i <= 1003;i += i&-i)
        for(int j = y;j <= 1003;j += j&-j)
            c[i][j] += val;
}

int ask(int x1,int y1,int x2,int y2)
{
    return sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1);
}

struct Node
{
    int a,c,d;
    ll cnt;
    Node(int _a=0,int _c=0,int _d=0,ll _e=0):a(_a),c(_c),d(_d),cnt(_e){}
    bool operator <(const Node &BB)const
    {
        if(a!=BB.a)return a<BB.a;
        if(c!=BB.c)return c<BB.c;
        return d<BB.d;
    }
}have[N];


int main(){
    scanf("%d",&t);
    while(t--)
    {
        F(i,1,100000)A[i]=0;
        memset(c,0,sizeof(c)),tot=0;
        scanf("%d%d",&n,&m);
        int a,b,c;
        F(i,1,n)
        {
            scanf("%d%d",&a,&b);
            if(a>A[b])A[b]=a,C[b]=0;
            if(a==A[b])C[b]++;
        }
        F(i,1,m)
        {
            scanf("%d%d%d",&a,&b,&c);
            if(A[c])have[++tot]=Node(A[c],a,b,C[c]);
        }
        sort(have+1,have+1+tot);
        int tcnt=1;
        F(i,2,tot)if(have[i].a==have[tcnt].a&&have[i].c==have[tcnt].c&&have[i].d==have[tcnt].d)
        {
            have[tcnt].cnt+=have[i].cnt;
        }else have[++tcnt]=have[i];
        ll ans=0;tot=tcnt;
        for(int i=tot;i;i--)
        {
            if(!ask(have[i].c,have[i].d,1000,1000))
                ans+=have[i].cnt;
            add(have[i].c,have[i].d,1);
        }
        printf("Case #%d: %lld
",++cas,ans);
    }
    return 0;
}