bzoj3192: [JLOI2013]删除物品(树状数组)

来源:互联网 发布:毕向东java全套视频 编辑:程序博客网 时间:2024/05/16 10:26

题目传送门
好题啊好思维。

解法:
亮哥过来问我这道题。
然后我就开始做了。。
想着想着这个人就过来跟我说把他们并起来就好了。
然后换顶的时候就相当于把分界线移动一格而已。
然后不就树状数组维护就好了。

哦还有就是本来三天前就A了可是没加longlong硬是WA。

两个桶头对头。
拿样例来说:
1
4
5
2
7
3
头对头。那么就是541273.
然后分界线在1的后面2的前面。
然后换头就相当于把分界线移动。
然后树状数组维护就好了呀。

代码实现:

#include<cstdio>#include<cstring>#include<cstdlib>#include<iostream>#include<algorithm>#include<cmath>#include<queue>using namespace std;typedef long long ll;struct node {ll x,s;}a[210000],b[110000];int n1,n2,n;bool cmp(node n1,node n2) {return n1.s>n2.s;}int lowbit(int x){return x&-x;}ll sum[110000];void change(int x,int k) {while(x<=n){sum[x]-=k;x+=lowbit(x);}}ll getsum(int x) {ll ans=0;while(x>0) {ans+=sum[x];x-=lowbit(x);}return ans;}int main() {    scanf("%d%d",&n1,&n2);    for(int i=1;i<=n1;i++)scanf("%lld",&a[n1-i+1].s);    for(int i=1;i<=n2;i++)scanf("%lld",&a[n1+i].s);n=n1+n2;    for(int i=1;i<=n;i++)a[i].x=i;sort(a+1,a+1+n,cmp);    memset(sum,0,sizeof(sum));for(int i=1;i<=n;i++)change(i,1);    ll f=n1,ans=0;    for(int i=1;i<=n;i++) {        if(a[i].x<=f) {ans+=abs(getsum(f)-getsum(a[i].x));f=a[i].x;change(a[i].x,-1);}        else {ans+=abs(getsum(a[i].x-1)-getsum(f));f=a[i].x;change(a[i].x,-1);}    }    printf("%lld\n",ans);    return 0;}
原创粉丝点击