P3658 [USACO17FEB]Why Did the Cow Cross the Road III P cdq分治

题意

两列n的全排列，相同的数连边，如果一对数有交叉且差的绝对值>k，则++ans，求ans

 

显然一个数字有三个元素  在a序列中的位置  在b序列中的位置  该数的权值

一开始想先在外部将权值排序好进去cdq  但是在cdq里面按照x排序的话权值就会乱  而权值有对判定起到作用 所以不好处理

可以将x排序完进入cdq分治很好处理

// luogu-judger-enable-o2
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(ll i=(a);i<=(b);i++)
#define repp(i,a,b) for(ll i=(a);i>=(b);--i)
#define ll long long
#define see(x) (cerr<<(#x)<<'='<<(x)<<endl)
#define inf 0x3f3f3f3f
#define CLR(A,v)  memset(A,v,sizeof A)
/////////////////////////////////////
const int N=1e6+10;
int n,m,t[N],k,ans[N];
void add(int x,int v){for(;x<=N;x+=x&-x)t[x]+=v;}
int qsum(int x){int ans=0;for(;x;x-=x&-x)ans+=t[x];return ans;}
struct node{int x,y,id;}s[N];
bool cmp(node a,node b){return a.y<b.y;}
void cdq(int l,int r)
{
    if(l==r)return ;int mid=(l+r)>>1;
    cdq(l,mid);cdq(mid+1,r);
    sort(s+l,s+mid+1,cmp);sort(s+mid+1,s+r+1,cmp);
    int j=mid;
    repp(i,r,mid+1)
    {
        while(j>=l&&s[j].y>=s[i].y)add(s[j].id,1),j--;
        
        if(s[i].id-k-1>=1)ans[s[i].id]+=qsum(s[i].id-k-1);
        if(s[i].id+k+1<=n)ans[s[i].id]+=qsum(n)-qsum(s[i].id+k);
    }
    while(++j<=mid)add(s[j].id,-1);
}

int main()
{
    cin>>n>>k;int x;
    rep(i,1,n)
    s[i].id=i,scanf("%d",&x),s[x].x=i;
    rep(i,1,n)
    scanf("%d",&x),s[x].y=i;
    sort(s+1,s+1+n,[](node a,node b){return a.x<b.x;});
    cdq(1,n);
    ll sum=0;
    rep(i,1,n)sum+=1ll*ans[i];
    cout<<sum;
    return 0;
}

View Code