ZOJ - 4053(2018 青岛网络赛) - 启发式 + 主席树

题目链接:点击这里

 

解题思路:

如果我们已经知道区间(l,r)的逆序对值W,现在对m进行标记,l<=m<=r.那么区间就被分为了两块[l,m),(m,r].

对于新得到的两个子区间,设区间长度小的那个子区间的长度为k,区间标记为rt

那么我们可以利用主席树在O(k*log(k))求出rt逆序对值,那么我们如何得到另一个区间的值呢？

设点i的正贡献为:[1,i-1]大于a[i]的个数,反贡献为[i+1,r]小于a[i]的贡献

很容易找到正贡献累加就是[1,n]的总逆序对个数,且正贡献总和于反贡献总和是相等的.

对于小的区间在左边,右区间的逆序对值 = W - (l,m)的反贡献和

对于小的区间在右边,左区间的逆序对值 = W - (m,r)的正贡献和

时间复杂度证明:

对于一个点i被暴力求出贡献值得情况是他在小的子区间,子区间大小<=原区间大小/2,所以i点最多被暴力log(n)次

所以总时间复杂度为O(n*log(n)*log(n))

 

#include<bits/stdc++.h>
#define lson l,mid
#define rson mid+1,r
using namespace std;
typedef long long ll;
int n,m;
const int mx = 1e5+10;
int a[mx],root[mx],sma[mx],big[mx];
int ls[20*mx],rs[20*mx],sum[20*mx],size;
ll pv[mx];
multiset <ll> mlst;
set <int> st;
void update(int x,int &y,int l,int r,int v,int M)
{
    y = ++size;
    ls[y] = ls[x],rs[y] = rs[x],sum[y] = sum[x] + v;
    int mid = (l+r)>>1;
    if(l==r) return ;
    if(M<=mid) update(ls[x],ls[y],lson,v,M);
    else update(rs[x],rs[y],rson,v,M);
}
int query1(int x,int l,int r,int M)
{
    if(l>M) return sum[x];
    int mid = (l+r)>>1;
    if(l==r) return 0;
    int ans = query1(rs[x],rson,M);
    if(M<mid) ans += query1(ls[x],lson,M);
    return ans;
}
int query2(int x,int l,int r,int M)
{
    if(r<M) return sum[x];
    int mid = (l+r)>>1;
    if(l==r) return 0;
    int ans = query2(ls[x],lson,M);
    if(M>mid+1) ans += query2(rs[x],rson,M);
    return ans;	
}
void init()
{	
    sum[0] = ls[0] = rs[0] = size = 0;
	mlst.clear();st.clear();
	mlst.insert(0);
	st.insert(0);st.insert(n+1);
}
void deal(int l,int mid,int r)
{
	int L = mid-l,R = r-mid;
	ll ret = 0,ans = 0;
	if(L||R){
		if(L<=R){
			for(int i=l+1;i<mid;i++)	
			ret += big[i]-query1(root[l-1],1,n,a[i]);
			if(L) mlst.insert(ret);
			for(int i=l;i<=mid;i++) 
			ans += query2(root[r],1,n,a[i])-sma[i];
			mlst.insert(ans = pv[l]-ans);
		}else{
			for(int i=mid+2;i<=r;i++)
			ans += big[i]-query1(root[mid],1,n,a[i]);
			if(R) mlst.insert(ans);
			for(int i=mid;i<=r;i++)
			ret += big[i]-query1(root[l-1],1,n,a[i]);
			mlst.insert(ret = pv[l]-ret);
		}
	}
	mlst.erase(mlst.find(pv[l]));
	pv[l] = ret,pv[mid+1] = ans;
}
int main()
{
    int t,cas = 1;
    scanf("%d",&t);
    while(t--){
		scanf("%d",&n);
		init();
		ll ans = 0;
		for(int i=1;i<=n;i++) scanf("%d",a+i);
		for(int i=1;i<=n;i++){
			update(root[i-1],root[i],1,n,1,a[i]);
			big[i] = query1(root[i-1],1,n,a[i]);
			sma[i] = query2(root[i-1],1,n,a[i]);
			ans += big[i];
		}
		mlst.insert(ans);
		pv[1] = ans;
		for(int i=1;i<=n;i++){
			ll ret = *(--mlst.end());
			scanf("%d",&m);
			printf("%lld%c",ret,i==n?'\n':' ');
			if(ret==0) continue;
			m = ret ^ m;
			auto it = st.lower_bound(m),ip = it;
			deal(*(--ip)+1,m,*it-1);
			st.insert(m);
		}
    }
    return 0;
}