HDU 5618 Jam's problem again(CDQ分治)

26 篇文章 0 订阅

Description
Jam喜欢坐标轴上的题,现在给出你一个三维的坐标轴,给出N个点,坐标分别为(x,y,z),如果有两个点(xi,yi,zi)和(xj,yj,zj),xi>=xj,yi>=yj,zi>=zj时,i点的等级就加一,每个等级一开始为0,当然1≤x,y,z≤100000 1≤N≤100000,现在求每个点的等级
Input
第一行T(1≤T≤15),表示TT组数据。
接下来TT组数据:
第一行为N,表示点的数目
接下来N行,每一行三个整数x,y,z表示点的坐标
Output
输出N行,每行表示每个点的等级
Sample Input
1
4
10 4 7
10 6 6
8 2 5
7 3 10
Sample Output
1
1
0
0
Solution
CDQ分治,以cnt[i]表示第i个点的级别,按第一维排序,CDQ(l,r)表示求出[l,r]中所有元素对[l,r]中任一元素的贡献值,那么问题核心转化为如何求[l,mid]中所有元素对[mid+1,r]中任一元素的贡献值,由于[l,mid]中所有元素的第一维都不大于[mid+1,r]中所有元素的第一维,对两个区间都以第二维为第一关键字排序,对于每个[mid+1,r]中的元素i,将[l,mid]中所有满足j.y<=i.y的元素以j.z为下标,1为键值插入到树状数组中,则只需统计树状数组中所有下标小于等于i.z的值之和累加到cnt[i]中即可
注意由于这个分治过程是有序的(按第一关键字分治),故如果两个元素的三个属性完全相同,那么在分治过程中编号靠前的元素的级别会少算,因为在计算其贡献的时候只考虑了编号在其前面的元素,故在分治前需要去重并统计每一个重复元素的数量,在分治的时候每次插入时的键值从1改成num[i]即可(num[i]表示去重后第i个点的数量)
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 111111
int T,n,tot,id[maxn];
struct node
{
    int x,y,z,cnt,ans,id;
}p[maxn];
int cmpx(node a,node b)
{
    if(a.x!=b.x)return a.x<b.x;
    if(a.y!=b.y)return a.y<b.y;
    return a.z<b.z;
}
int cmpy(node a,node b)
{
    if(a.y!=b.y)return a.y<b.y;
    return a.z<b.z;
}
int cmpid(node a,node b)
{
    return a.id<b.id;
}
struct BIT 
{
    #define lowbit(x) (x&(-x))
    int b[maxn];
    void init()
    {
        memset(b,0,sizeof(b));
    }
    void update(int x,int v)
    {
        while(x<=maxn)
        {
            b[x]+=v;
            x+=lowbit(x);
        }
    }
    int query(int x)
    {
        int ans=0;
        while(x)
        {
            ans+=b[x];
            x-=lowbit(x);
        }
        return ans;
    }
    void clear(int x)
    {
        while(x<=maxn)
        {
            b[x]=0;
            x+=lowbit(x);
        }
    }
}bit;
void CDQ(int l,int r)
{
    if(l==r)
    {
        p[l].ans+=p[l].cnt-1;
        return ;
    }
    int mid=(l+r)>>1; 
    CDQ(l,mid);
    CDQ(mid+1,r);
    sort(p+l,p+mid+1,cmpy);
    sort(p+mid+1,p+r+1,cmpy);
    int j=l;
    for(int i=mid+1;i<=r;i++)
    {
        for(;j<=mid&&p[j].y<=p[i].y;j++)
            bit.update(p[j].z,p[j].cnt);
        p[i].ans+=bit.query(p[i].z);
    }
    for(int i=l;i<j;i++)bit.update(p[i].z,-p[i].cnt);
    sort(p+l,p+r+1,cmpy);
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        bit.init();tot=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z),p[i].ans=0,p[i].id=i;   
        sort(p+1,p+n+1,cmpx);
        for(int i=1;i<=n;i++)
        {
            if(i!=1&&p[i-1].x==p[i].x&&p[i-1].y==p[i].y&&p[i-1].z==p[i].z)
                p[tot].cnt++;
            else p[++tot]=p[i],p[tot].cnt=1;
            id[p[i].id]=tot;
            p[tot].id=tot;  
        }
        CDQ(1,tot);
        sort(p+1,p+tot+1,cmpid);
        for(int i=1;i<=n;i++)printf("%d\n",p[id[i]].ans);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值