cdq样例不过求调
查看原帖
cdq样例不过求调
1491240
abnegate楼主2025/7/30 15:42
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+5;
int n,m;
int pos[N];
////////////////////////////////////////////
struct node{
	int t=0;//插入时间维  1	
	int id;//下标维     2
	int val;//权值维    3
	int res;
}a[N];
bool cmpt(node x,node y){
	if(x.t!=y.t) return x.t<y.t;
	if(x.id!=y.id) return x.id<y.id;
	return x.val<y.val;
}
bool cmpid(node x,node y){
	if(x.id!=y.id) return x.id<y.id;
	return x.val<y.val;
}
////////////////////////////////////////////
int sum[N];
int lowbit(int x){
	return x&-x;
}
void update(int x,int y){
	while(x<=n+1){
		sum[x]+=y;
		x+=lowbit(x);
	}
}
int query(int x){
	int res=0;
	while(x>=1){
		res+=sum[x];
		x-=lowbit(x);
	}
	return res;
}
////////////////////////////////////////////
void cdq(int l,int r){
	if(l==r) return;
	int mid=(l+r)>>1;
	cdq(l,mid);
	cdq(mid+1,r);
	sort(a+l,a+mid+1,cmpid);
	sort(a+mid+1,a+r+1,cmpid);
	int i=l,j=mid+1;
	while(j<=r){
		while(i<=mid&&a[i].id<a[j].id){
			update(a[i].val,1);
			i++;
		}
		a[j].res+=query(n)-query(a[j].val);
		j++;
	}
	for(int k=l;k<i;k++) update(a[k].val,-1);
	i=l,j=mid+1;
	while(j<=r){
		while(i<=mid&&a[i].id>a[j].id){
			update(a[i].val,1);
			i++;
		}
		a[j].res+=query(a[j].val);
		j++;
	}
	for(int k=l;k<i;k++) update(a[k].val,-1);
}
////////////////////////////////////////////
signed main()
{
	std::ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i].val;
		a[i].id=i;
		pos[a[i].val]=i;
	}
	for(int i=1,x;i<=m;i++){
		cin>>x;
		a[pos[x]].t=m-i+1;
	}
	sort(a+1,a+1+n,cmpt);
	cdq(1,n);
	sort(a+1,a+1+n,cmpt);
	for(int i=1;i<=n;i++){
		a[i].res+=a[i-1].res;
	}
	for(int i=n;i>=n-m+1;i--){
		cout<<a[i].res<<'\n';
	}
	
	return 0;
}

2025/7/30 15:42
加载中...