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

来源:互联网 发布:mac双系统移除win8后 编辑:程序博客网 时间:2024/05/22 14:55

题意:箱子再分配问题需要解决如下问题:
(1)一共有N个物品,堆成M堆。
(2)所有物品都是一样的,但是它们有不同的优先级。
(3)你只能够移动某堆中位于顶端的物品。
(4)你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的,可以直接将之删除而不用移动。

(5)求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
(6)只是一个比较难解决的问题,这里你只需要解决一个比较简单的版本:
不会有两个物品有着相同的优先级,且M=2

看完题觉得是水题,以为xjb模拟就能过,无脑了一发果然T了,分析了一下模拟是n^2的,然后感觉自己歇菜了。
想了想感觉好像用个数据结构维护就可以了,把第一个倒序第二个正序拼在一起,然后排个序,看看和原序列的差是多少,注意每一次计算完以后要把贡献删除不然会算重。
建议手玩一波样例就理解了。

#include<cstdio>#include<algorithm>#include<cstring>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fd(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=3e5+5;int a1[N],a2[N],n1,n2,n;int t[N],t2[N],tot;struct node{    int x,id;}a[N];bool cmp(node a,node b){    return a.x>b.x;}inline int lowbit(int x){    return x&(-x);}inline void add(int x,int y){    while (x<=n)    {        t[x]+=y;        x+=lowbit(x);    }}inline int ask(int x){    int ans=0;    while (x>0)    {        ans+=t[x];        x-=lowbit(x);    }    return ans;}int main(){    scanf("%d%d",&n1,&n2);    fd(i,n1,1)    scanf("%d",&a[i].x),a[i].id=i;    fo(i,n1+1,n1+n2)    scanf("%d",&a[i].x),a[i].id=i;    n=n1+n2;    sort(a+1,a+1+n,cmp);    fo(i,1,n)add(i,1);    int x=n1,ans=0;    fo(i,1,n)    {        if (a[i].id>x)ans+=ask(a[i].id-1)-ask(x);        else ans+=ask(x)-ask(a[i].id);        add(a[i].id,-1);        x=a[i].id;    }    printf("%d\n",ans);}
原创粉丝点击