题意
两列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; }