【bzoj3333】排队计划 （线段树）

3333: 排队计划

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 717   Solved: 323
[ Submit][ Status][ Discuss]

Description

Input

Output

Sample Input

6 2
160 163 164 161 167 160
2
3

Sample Output

6
3
1

HINT

【题目思路】

每次操作是把一个数后面所有小于等于该数值的位置都排序重新放置到原位置，那么一次操作减少的逆序对个数就是该次操作调整的所有数构成的逆序对个数（就是每个点后面小于自己的数的个数的和）。

下面有几点性质：

1、每次操作以后，所有统计过的点都不用再统计（且下次若是要操作这个点也不用操作），因为如果h[i]已经被操作过,并且到了j的位置，成了h1[j](即h1[j]==h[i])，那么h1[j]后面的值肯定都是比h1[j]要大的，逆序对个数肯定不会改变。 若是h1[j]后面的值h1[k]是和h1[j]在同一操作中调整过的，那么h1[k]>=h1[j],若是没有调整过，那么说明该值肯定大于h1[j](没被调整过说明  该值>那次调整选择的值>=h1[j])。

2、对于一个没被操作过的位置，无论后面的数如何操作，对于这个数而言这个数与后面的所有数构成的逆序对的个数不变。

所以每个位置都只会统计一次，预处理出所有的位置与后面的数构成的逆序对个数，然后每次操作，把统计过的点都标记，每次操作的时候只用统计没被标记过的点。

用一个线段树就可以处理。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 500005
#define ll long long
#define INF 1e9+7
using namespace std;
int tr[N*5],f[N],mark[N*5],n,m,h[N],x,mx=-INF;
ll ans=0;
struct he{
	int d,p;
}a[N];
void insert(int p,int l,int r,int x){
	if(x>r||l>x) return ;
	if(l==x&&r==x) {
		tr[p]++;
		return ;
	}
	int mid=(l+r)/2;
	insert(p*2,l,mid,x);insert(p*2+1,mid+1,r,x);
	tr[p]=tr[p*2]+tr[p*2+1];
}
int find(int p,int l,int r,int x,int y){
	if(y<l||r<x) return 0;
	if(x<=l&&r<=y) return tr[p];
	int mid=(l+r)/2;
	return find(p*2,l,mid,x,y)+find(p*2+1,mid+1,r,x,y);
}
void build(int p,int l,int r){
	mark[p]=1;
	if(l==r) {
		tr[p]=h[l];
		return ;
	}
	int mid=(l+r)/2;
	build(p*2,l,mid);build(p*2+1,mid+1,r);
	tr[p]=min(tr[p*2],tr[p*2+1]);
}
void modify(int p,int l,int r,int x,int y,int u){
	if(x>r||l>y) return ;
	if(mark[p]==0) return ;
	if(tr[p]>u) return ;
	if(l==r){
		mark[p]=0;
		tr[p]=INF;
		ans-=f[l];
		return ;
	}
	int mid=(l+r)/2;
	modify(p*2,l,mid,x,y,u);modify(p*2+1,mid+1,r,x,y,u);
	mark[p]=mark[p*2]|mark[p*2+1];
	tr[p]=min(tr[p*2],tr[p*2+1]);
}
bool cmp(he a, he b){
	return a.d<b.d;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i].d),a[i].p=i;
	sort(a+1,a+1+n,cmp);
	mx=1;h[a[1].p]=1;
	for(int i=2;i<=n;i++) {
		if(a[i].d!=a[i-1].d) mx++;
		h[a[i].p]=mx;
	}
	for(int i=n;i>=1;i--){
		f[i]=find(1,1,mx,1,h[i]-1);
		insert(1,1,mx,h[i]);
		ans+=f[i];
	}
//	memset(tr,0,sizeof(tr));
	build(1,1,n);
	printf("%lld\n",ans);
	for(int i=1;i<=m;i++){
		scanf("%d",&x);
		if(find(1,1,n,x,x)==INF) {printf("%lld\n",ans);continue;}
		modify(1,1,n,x,n,h[x]);
		printf("%lld\n",ans);
	}
}