bzoj3192 删除物品 树状数组

       由于m=2就变成了水题一道。。。(整天水题刷刷没救了)

       可以把两个栈变成一个数组,然后就变成了每次从当前最大的移动到第二大的,统计路径上还有多少个点累加入答案,然后把最大的删除。直接排序后用树状数组维护前缀和即可。

AC代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#define ll long long
using namespace std;

int n,m,c[100005]; struct node{ int x,id; }a[100005];
void del(int x){
	for (; x<=n; x+=x&-x) c[x]--;
}
int getsum(int x){
	int t=0; for (; x; x-=x&-x) t+=c[x]; return t;
}
bool cmp(node u,node v){ return u.x>v.x; }
int main(){
	scanf("%d%d",&n,&m); int i; a[0].id=n;
	for (i=n; i; i--){ scanf("%d",&a[i].x);  a[i].id=i; }
	for (i=1; i<=m; i++){
		scanf("%d",&a[++n].x); a[n].id=n;
	}
	sort(a+1,a+n+1,cmp);
	for (i=1; i<=n; i++) c[i]=i&-i;
	ll ans=0;
	for (i=1; i<=n; i++){
		if (a[i].id>a[i-1].id) ans+=getsum(a[i].id-1)-getsum(a[i-1].id);
			else ans+=getsum(a[i-1].id)-getsum(a[i].id);
		del(a[i].id);
	}
	printf("%lld\n",ans);
}


by lych

2016.3.14

你可能感兴趣的:(树状数组)