[BZOJ4430][Nwerc2015]Guessing Camels赌骆驼(cdq分治)

题目:

我是超链接

题解:

看上去毫无头绪,但字里行间的“相对位置”很有意思,代表了每头骆驼在三个人中的位置的“相对大小”
那么只需要把每头骆驼在三个人中的位置建立一个三元组(x,y,z),就是一个很裸的cdq分治啦

代码:

#include 
#include 
#define LL long long
using namespace std;
const int N=200005;
struct hh{int x,y,z;LL ans;}f[N],jl[N];
int c[N],n;
void add(int loc,int val){for (int i=loc;i<=n;i+=i&(-i)) c[i]+=val;}
int qurry(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i)) ans+=c[i];
    return ans;
}
void cdq(int l,int r)
{
    if (l>=r) return;
    int mid=(l+r)>>1;
    cdq(l,mid); cdq(mid+1,r);
    int t1=l,t2=mid+1;
    for (int i=l;i<=r;i++)
      if ((t1<=mid && f[t1].y<=f[t2].y) || t2>r) jl[i]=f[t1++];
      else jl[i]=f[t2++];

    for (int i=l;i<=r;i++)
      if ((f[i]=jl[i]).x<=mid) add(f[i].z,1);
      else f[i].ans+=qurry(f[i].z);
    for (int i=l;i<=r;i++) if (f[i].x<=mid) add(f[i].z,-1);
}
int cmp(hh a,hh b){return a.xx;}
int main()
{
    int a;
    scanf("%d",&n);
    for (int i=1;i<=n;i++){scanf("%d",&a);f[a].x=i;}
    for (int i=1;i<=n;i++){scanf("%d",&a);f[a].y=i;}
    for (int i=1;i<=n;i++){scanf("%d",&a);f[a].z=i;}
    sort(f+1,f+n+1,cmp);
    cdq(1,n);LL ans=0;
    for (int i=1;i<=n;i++) ans+=f[i].ans;
    printf("%lld",ans);
}

你可能感兴趣的:(CDQ分治/整体二分)