【分块+BIT】BZOJ3787 Gty的文艺妹子序列

【题目】
BZOJ
正整数序列 A A A,支持单点修改,区间逆序对数,所有数字不超过 5 × 1 0 4 5\times 10^4 5×104。强制在线。

【解题思路】
毒瘤数据结构题不妨考虑分块,设分块大小为 S S S

维护 s u m i , j sum_{i,j} sumi,j表示前 i i i块中元素 j j j的个数,每次修改只会影响 O ( S ) O(S) O(S)个值,同样用树状数组维护可以做到 O ( S log ⁡ n ) O(S\log n) O(Slogn)
由于我们询问时需要对第二维进行区间查询,所以第二维要用树状数组维护起来。

维护 a n s i , j ans_{i,j} ansi,j表示第 i i i块和第 j j j块之间元素组成的逆序对个数,每次修改只会影响 O ( n S ) O(\frac n S) O(Sn)个值,利用上面维护的 s u m sum sum可以快速进行更新,做到 O ( n S log ⁡ n ) O(\frac n S \log n) O(Snlogn)
由于我们询问时需要对第二维进行区间查询,所以第二维要用树状数组维护起来。

最后维护 n u m i num_i numi表示第 i i i块内逆序对个数,每次修改可以 O ( S ) O(S) O(S)暴力算。

询问时分成四部分:块内统计 n u m i num_i numi可以做到 O ( n S ) O(\frac n S) O(Sn),块间通过维护 a n s i , j ans_{i,j} ansi,j第二维的树状数组可以做到 O ( n S log ⁡ S ) O(\frac n S \log S) O(SnlogS),散块和整块之间可以通过维护 s u m i , j sum_{i,j} sumi,j第二维的树状数组可以做到 O ( S log ⁡ S ) O(S\log S) O(SlogS),散块部分可以暴力维护做到 O ( S log ⁡ n ) O(S\log n) O(Slogn)

S = n S=\sqrt n S=n 时可以取到最优复杂度 O ( n n log ⁡ n ) O(n\sqrt n \log n) O(nn logn)

想清楚了很好写。所以一点也不毒瘤。
【参考代码】

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N=5e4+10,L=233;

namespace IO
{
	int read()
	{
		int ret=0;char c=getchar();
		while(!isdigit(c)) c=getchar();
		while(isdigit(c)) ret=ret*10+(c^48),c=getchar();
		return ret;
	}
	void write(int x){if(x>9)write(x/10);putchar(x%10^48);}
	void writeln(int x){write(x);putchar('\n');}
}
using namespace IO;

namespace BITisgood
{
#define lowbit(x) (x&(-x))
	struct BIT
	{
		vector<int>c;int lim;
		void init(int x){c.clear();c.resize(x+1,0);lim=x;}
		void update(int x,int v){for(;x<=lim;x+=lowbit(x))c[x]+=v;}
		int qry(int x){int res=0;for(;x;x-=lowbit(x))res+=c[x];return res;}
		int query(int l,int r){return l>r?0:qry(r)-qry(l-1);}
	};
#undef lowbit
}
using namespace BITisgood;

namespace DreamLolita
{
	int n,Q,cnt,a[N],bl[N];
	ll fans;
	struct Block
	{
		int num,st,ed;
	}B[N/L+10];
	BIT sum[N/L+10],ans[L+10],tmp;//wrong because ans[L],wrong memory!
	ll Query(int l,int r)
	{
		ll res=0;
		if(bl[l]==bl[r]) 
		{
			for(int i=l;i<=r;++i) res+=tmp.query(a[i]+1,n),tmp.update(a[i],1);
			for(int i=l;i<=r;++i) tmp.update(a[i],-1);
			return res;
		}
		for(int i=bl[l]+1;i<bl[r];++i) 
		{
			res+=B[i].num;
			res+=ans[i].query(i+1,bl[r]-1);
		}
		if(bl[l]+1<=bl[r]-1)
		{
			for(int i=l;i<=B[bl[l]].ed;++i) res+=sum[bl[r]-1].query(1,a[i]-1)-sum[bl[l]].query(1,a[i]-1);
			for(int i=B[bl[r]].st;i<=r;++i) res+=sum[bl[r]-1].query(a[i]+1,n)-sum[bl[l]].query(a[i]+1,n);
		}
		for(int i=l;i<=B[bl[l]].ed;++i) res+=tmp.query(a[i]+1,n),tmp.update(a[i],1);
		for(int i=B[bl[r]].st;i<=r;++i) res+=tmp.query(a[i]+1,n),tmp.update(a[i],1);
		for(int i=l;i<=B[bl[l]].ed;++i) tmp.update(a[i],-1);
		for(int i=B[bl[r]].st;i<=r;++i) tmp.update(a[i],-1);
		return res;
	}
	void Update(int p,int v)
	{
		for(int i=1;i<bl[p];++i) ans[i].update(bl[p],-(sum[i].query(a[p]+1,n)-sum[i-1].query(a[p]+1,n)));
		for(int i=bl[p]+1;i<=cnt;++i) ans[bl[p]].update(i,-(sum[i].query(1,a[p]-1)-sum[i-1].query(1,a[p]-1)));
		for(int i=bl[p];i<=cnt;++i) sum[i].update(a[p],-1),sum[i].update(v,1); a[p]=v;
		for(int i=1;i<bl[p];++i) ans[i].update(bl[p],(sum[i].query(a[p]+1,n)-sum[i-1].query(a[p]+1,n)));
		for(int i=bl[p]+1;i<=cnt;++i) ans[bl[p]].update(i,(sum[i].query(1,a[p]-1)-sum[i-1].query(1,a[p]-1)));
		B[bl[p]].num=0;
		for(int i=B[bl[p]].st;i<=B[bl[p]].ed;++i) B[bl[p]].num+=tmp.query(a[i]+1,n),tmp.update(a[i],1);
		for(int i=B[bl[p]].st;i<=B[bl[p]].ed;++i) tmp.update(a[i],-1);
	}
	void solution()
	{
		n=read();cnt=(n-1)/L+1;
		for(int i=1;i<=n;++i) a[i]=read(),bl[i]=(i-1)/L+1,B[bl[i]].ed=i;
		for(int i=n;i;--i) B[bl[i]].st=i;
		for(int i=0;i<=cnt;++i) sum[i].init(n),ans[i].init(cnt);
		tmp.init(n);
		for(int i=1;i<=cnt;++i) 
		{
			for(int j=0;j<=n;++j) sum[i].c[j]=sum[i-1].c[j];
			for(int j=B[i].st;j<=B[i].ed;++j) 
			{
				sum[i].update(a[j],1);
				for(int k=1;k<i;++k) ans[k].update(i,sum[k].query(a[j]+1,n)-sum[k-1].query(a[j]+1,n));
				for(int k=B[i].st;k<j;++k) if(a[k]>a[j]) B[i].num++;
			}
		}
		Q=read();
		while(Q--)
		{
			int op=read(),x=read()^fans,y=read()^fans;
			if(op) Update(x,y);
			else printf("%lld\n",(fans=Query(x,y)));
		}
	}
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("BZOJ3787.in","r",stdin);
	freopen("BZOJ3787.out","w",stdout);
#endif
	DreamLolita::solution();
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值