【hdu 5517】【2015ACM/ICPC亚洲区沈阳站】Triple 题意&题解&代码(C++)

来源:互联网 发布:360软件管家官方下载 编辑:程序博客网 时间:2024/05/02 15:01

题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=5517
题意:
给出n个二元组<a,b>和m个三元组<c,d,e>现在将其组合成一个新的三元组集合,新三元组中允许有重复元素,一个二元组和一个三元组可以组合当且仅当b==e时将其组合成<a,c,d>,在新组合的三元组中任取一元素<a,b,c>,如果在新组合存在一个元素<u,v,w>使得u!=a && v!=b && w!=c 同时 u>=a && v>=b &&w>=c 则元素<a,b,c>不统计在答案的集合中,否则统计在答案的集合中,问最终答案的集合中有多少元素,可以有重复元素。
题解:
有三维信息,我们发现新三元组中a的值在10^5范围,而b,c范围都在1000以内,这样的话可以用a排序,然后用b,c统计在二维树状数组中,在组合新的三元组时也有一些技巧和方法,可以参考代码来理解。
本题要用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);    }}
0 0
原创粉丝点击