【hdu 5517】【2015ACM/ICPC亚洲区沈阳站】Triple 题意＆题解＆代码（Ｃ＋＋）

题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=5517
题意：
给出ｎ个二元组<ａ，ｂ>和ｍ个三元组<ｃ，ｄ，ｅ>现在将其组合成一个新的三元组集合，新三元组中允许有重复元素，一个二元组和一个三元组可以组合当且仅当ｂ==ｅ时将其组合成<ａ，ｃ，ｄ>，在新组合的三元组中任取一元素<ａ，ｂ，ｃ>，如果在新组合存在一个元素<ｕ，ｖ，ｗ>使得u!=a && v!=b && w!=c 同时 u>=a && v>=b &&w>=c　则元素<ａ，ｂ，ｃ>不统计在答案的集合中，否则统计在答案的集合中，问最终答案的集合中有多少元素，可以有重复元素。
题解：
有三维信息，我们发现新三元组中ａ的值在10^5范围，而b，c范围都在1000以内，这样的话可以用ａ排序，然后用ｂ，ｃ统计在二维树状数组中，在组合新的三元组时也有一些技巧和方法，可以参考代码来理解。
本题要用long long，多组数据注意初始化。
代码：

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
using namespace std;
int T,n,m;
int id[100005];
long long cnt[100005];
struct node{
    int a;int b;int c;long long num;
}p[100005],np[100005];
long long sum[1005][1005];
int cmp(node x1,node x2)
{
    if (x1.a!=x2.a) return x1.a<x2.a;
    if (x1.b!=x2.b) return x1.b<x2.b;
    return x1.c<x2.c;  
}
int pan(int i)
{
    if (p[i].a==p[i+1].a&&p[i].b==p[i+1].b&&p[i].c==p[i+1].c)
    return 1;
    return 0;
}
int lowbit(int x)
{
    return x&(-x);
}
int add(int x,int y,long long v)
{
    for (int i=x;i<=1000;i+=lowbit(i))
    for (int j=y;j<=1000;j+=lowbit(j))
    sum[i][j]+=v;   
}
long long query(int x,int y)
{
    long long res=0;
    for (int i=x;i>=1;i-=lowbit(i))
    for (int j=y;j>=1;j-=lowbit(j))
    res+=sum[i][j];
    return res;

}
int solve(int x,int y)
{
    long long res=query(1000,1000)-query(x-1,1000)-query(1000,y-1)+query(x-1,y-1);
    if (res==0)
    return 1;
    else
    return 0;
}
int main()
{
    scanf("%d",&T);
    for (int t=1;t<=T;t++)
    {
        scanf("%d%d",&n,&m);
        memset(sum,0,sizeof(sum));
        memset(id,0,sizeof(id));
        memset(cnt,0,sizeof(cnt));
        for (int i=1;i<=n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if (id[b]<a)
            {
                id[b]=a;
                cnt[b]=1;
            }
            else if (id[b]==a)
            cnt[b]++;
        }
        int tot=0;
        for (int i=1;i<=m;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            if (id[c]!=0)
            {
                tot++;
                p[tot].a=id[c];
                p[tot].b=a;
                p[tot].c=b;
                p[tot].num=cnt[c];
            }
        }
        sort(p+1,p+1+tot,cmp);
        int nn=0;
        long long res=0;
        for (int i=1;i<=tot;i++)
        {
            if (pan(i))
            res+=p[i].num;
            else
            {
                nn++;
                np[nn].num=0;               
                np[nn]=p[i];
                np[nn].num+=res;
                res=0;
            }
        }
        long long ans=0;
        for (int i=nn;i>=1;i--)
        {
            if (solve(np[i].b,np[i].c)) ans+=np[i].num;
            add(np[i].b,np[i].c,np[i].num);
        }
        printf("Case #%d: %I64d\n",t,ans);
    }
}