[BZOJ 4415][Shoi2013]发牌(树状数组+二分)

Description

假设一开始,荷官拿出了一副新牌,这副牌有N张不同的牌,编号依次为1到N。由于是新牌,所以牌是按照顺序排好的,从牌库顶开始,依次为1, 2,……直到N,N号牌在牌库底。为了发完所有的牌,荷官会进行N次发牌操作,在第i次发牌之前,他会连续进行R_i次销牌操作,R_i由输入给定。请问最后玩家拿到这副牌的顺序是什么样的?

举个例子,假设N = 4,则一开始的时候,牌库中牌的构成顺序为{1, 2, 3, 4}。

假设R1=2,则荷官应该连销两次牌,将1和2放入牌库底,再将3发给玩家。目前牌库中的牌顺序为{4, 1, 2}。

假设R2=0,荷官不需要销牌,直接将4发给玩家,目前牌库中的牌顺序为{1,2}。

假设R3=3,则荷官依次销去了1, 2, 1,再将2发给了玩家。目前牌库仅剩下一张牌1。

假设R4=2,荷官在重复销去两次1之后,还是将1发给了玩家,这是因为1是牌库中唯一的一张牌。

Solution

学习了一下怎么在树状数组上二分…

大概就是查询完第k大的数(每次洗牌的时候算一下这个k)之后就把这个数删掉

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define MAXN 700005
using namespace std;
int n,c[MAXN];
int read()
{
    int x=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){
        if(c=='-')f=-1;c=getchar();
    }
    while(c>='0'&&c<='9'){
        x=x*10+c-'0';c=getchar();
    }
    return x*f;
}
int lowbit(int x){return x&-x;}
void build()
{
    for(int i=1;i<=n;i++)
    {
        c[i]++;
        if(i+lowbit(i)<=n)c[i+lowbit(i)]+=c[i];
    }
}
int solve(int sum)
{
    int x=0,now=0;
    for(int i=(1<<22);i;i>>=1)
    {
        if(x+i<=n&&now+c[x+i]<sum)
        now+=c[x+i],x+=i;
    }
    ++x;
    for(int i=x;i<=n;i+=lowbit(i))c[i]--;
    return x;
}
int query(int pos)
{
    int res=0;
    while(pos>0)
    {res+=c[pos];pos-=lowbit(pos);}
    return res;
}
int main()
{
    n=read();build();
    int now=1;
    for(int i=n;i>0;i--)
    {
        int r=read();
        now+=r,now%=i;
        if(!now)now+=i;
        printf("%d\n",solve(now));
    }
    return 0;
} 

 

posted @ 2017-05-18 20:04  Zars19  阅读(337)  评论(1编辑  收藏  举报