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

  既然要从一个堆的堆顶按顺序拿出来放到第二个堆的堆顶,那么我们就可以把两个堆顶怼在一起,这样从一个堆拿到另一个堆只需要移动指针就好了。

  换句话说,把1~n倒着,n+1到n+m正着,用一个指针把两个序列分开就好了,然后用树状数组维护哪些点被取过就好了...

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=500010;
int n, m;
ll ans;
int a[maxn], b[maxn], pos[maxn], tree[maxn];
void read(int &k)
{
    int f=1; k=0; char c=getchar();
    while(c<'0' || c>'9') c=='-' && (f=-1), c=getchar();
    while(c<='9' && c>='0') k=k*10+c-'0', c=getchar();
    k*=f;
}
inline void add(int x, int delta){for(;x<=n;x+=x&-x) tree[x]+=delta;}
inline int query(int x){int sum=0; for(;x;x-=x&-x) sum+=tree[x]; return sum;}
int main()
{
    read(n); read(m);
    for(int i=n;i;i--) read(a[i]), b[i]=a[i];
    for(int i=n+1;i<=n+m;i++) read(a[i]), b[i]=a[i];
    n+=m; for(int i=1;i<=n;i++) tree[i]=i&-i;
    sort(b+1, b+1+n);
    for(int i=1;i<=n;i++) a[i]=lower_bound(b+1, b+1+n, a[i])-b;
    for(int i=1;i<=n;i++) pos[a[i]]=i;
    int now=n-m;
    for(int i=n;i;i--)
    {
        if(pos[i]>now) ans+=query(pos[i]-1)-query(now);
        else ans+=query(now)-query(pos[i]);
        add(pos[i], -1); now=pos[i];
    }
    printf("%lld\n", ans);
}
View Code

  貌似还可以用平衡树做,等会试试。

转载于:https://www.cnblogs.com/Sakits/p/8017923.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值