【BZOJ4889】[Tjoi2017]不勤劳的图书管理员 分块+树状数组

【BZOJ4889】[Tjoi2017]不勤劳的图书管理员

题目描述

加里敦大学有个帝国图书馆，小豆是图书馆阅览室的一个书籍管理员。他的任务是把书排成有序的，所以无序的书让他产生厌烦，两本乱序的书会让小豆产生这两本书页数的和的厌烦度。现在有n本被打乱顺序的书，在接下来m天中每天都会因为读者的阅览导致书籍顺序改变位置。因为小豆被要求在接下来的m天中至少要整理一次图书。小豆想知道，如果他前i天不去整理，第i天他的厌烦度是多少，这样他好选择厌烦度最小的那天去整理。

输入输出格式

输入格式：

第一行会有两个数，n，m分别表示有n本书，m天

接下来n行，每行两个数，ai和vi，分别表示第i本书本来应该放在ai的位置，这本书有vi页，保证不会有放置同一个位置的书

接下来m行，每行两个数，xj和yj，表示在第j天的第xj本书会和第yj本书会因为读者阅读交换位置

输出格式：

一共m行，每行一个数，第i行表示前i天不去整理，第i天小豆的厌烦度，因为这个数可能很大，所以将结果模10^9 +7后输出

输入输出样例

输入样例#1：

5 5
1 1
2 2
3 3
4 4
5 5
1 5
1 5
2 4
5 3
1 3

输出样例#1：

42
0
18
28
48

说明

对于20%的数据，1 ≤ ai; xj; yj ≤ n ≤ 5000, m ≤ 5000, vi ≤ 10^5

对于100%的数据，1 ≤ ai; xj; yj ≤ n ≤ 50000, m ≤ 50000, vi ≤ 10^5

题解：其实就是让你求一个区间中的带权逆序对数，依然用分块。对于每个块，维护两个树状数组s1,s2。s1代表书的个数，s2代表书的页数的前缀和。修改的时候，只有l,r和(l,r)中的数间的逆序对会改变，两边的不会改变。所以先暴力判断连边的小块，再扫中间的大块。设块的大小为B，假设一个块中有num本书的a值比l小，这些数的页数和为sum，那么ans-=sum-(B-num)*v[l]。r类似。

结果一交上去TLE了，实测109s多，改了一下块的大小就60s了。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn=50010;
const ll mod=1000000007;
ll ans;
int n,m,B;
ll s[2][250][maxn],v[maxn];
int p[maxn];
int rd()
{
	int ret=0;	char gc=getchar();
	while(gc<'0'||gc>'9')	gc=getchar();
	while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
	return ret;
}
void updata(int z,int y,int x,ll val)
{
	if(!x)	return ;
	for(int i=x;i<=n;i+=i&-i)	s[z][y][i]=(s[z][y][i]+val+mod)%mod;
}
ll query(int z,int y,int x)
{
	if(x<0)	return 0;
	ll ret=0;
	for(int i=x;i;i-=i&-i)	ret=(ret+s[z][y][i])%mod;
	return ret;
}
void calc(int a,int b,int c)
{
	if(p[a]<p[c])	ans=(ans+v[a]+v[c])%mod;
	else	ans=(ans-v[a]-v[c]+mod+mod)%mod;
	if(p[b]<p[c])	ans=(ans-v[b]-v[c]+mod+mod)%mod;
	else	ans=(ans+v[b]+v[c])%mod;
}
int main()
{
	int i,j,a,b,c,d;
	n=rd(),m=rd(),B=int(sqrt(n*17));
	for(i=0;i<n;i++)	p[i]=rd(),v[i]=rd(),updata(0,i/B,p[i],v[i]),updata(1,i/B,p[i],1);
	for(i=n-1;i>=0;i--)
	{
		ans=(ans+query(0,n/B+1,p[i])+query(1,n/B+1,p[i])*v[i])%mod;
		updata(0,n/B+1,p[i],v[i]),updata(1,n/B+1,p[i],1);
	}
	for(i=1;i<=m;i++)
	{
		a=rd()-1,b=rd()-1;
		if(a==b)
		{
			printf("%lld\n",ans);
			continue;
		}
		if(a>b)	swap(a,b);
		c=a/B,d=b/B;
		if(p[a]<p[b])	ans=(ans+v[a]+v[b])%mod;
		else	ans=(ans-v[a]-v[b]+mod+mod)%mod;
		if(c==d)
		{
			for(j=a+1;j<b;j++)	calc(a,b,j);
			swap(p[a],p[b]),swap(v[a],v[b]);
			printf("%lld\n",ans);
			continue;
		}
		for(j=a+1;j<c*B+B;j++)	calc(a,b,j);
		for(j=d*B;j<b;j++)	calc(a,b,j);
		for(j=c+1;j<d;j++)
		{
			ans=(ans-2*query(0,j,p[a])+query(0,j,n)-(2*query(1,j,p[a])-B+mod)*v[a]%mod+mod)%mod;
			ans=(ans+2*query(0,j,p[b])-query(0,j,n)+(2*query(1,j,p[b])-B+mod)*v[b]%mod+mod)%mod;
		}
		updata(0,c,p[a],-v[a]),updata(0,d,p[b],-v[b]),updata(1,c,p[a],-1),updata(1,d,p[b],-1);
		swap(p[a],p[b]),swap(v[a],v[b]);
		updata(0,c,p[a],v[a]),updata(0,d,p[b],v[b]),updata(1,c,p[a],1),updata(1,d,p[b],1);
		printf("%lld\n",ans);
	}
	return 0;
}
//5 5 1 1 2 2 3 3 4 4 5 5 1 5 1 5 2 4 5 3 1 3