bzoj 4415: [Shoi2013]发牌

最新推荐文章于 2019-09-24 18:38:59 发布

fyc_kabuto

最新推荐文章于 2019-09-24 18:38:59 发布

阅读量291

点赞数 1

分类专栏：线段树

本文链接：https://blog.csdn.net/qq_36808030/article/details/78165879

版权

线段树专栏收录该内容

23 篇文章 0 订阅

订阅专栏

题意：

约瑟夫问题。

题解：

约瑟夫问题……
直接线段树，记录下当前在哪一个，在线段树上二分下第k大就好了。
因为没想好再打，代码丑。
code：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
using namespace std;
struct trnode{
    int lc,rc,c;
}tr[1400010];int tot=0,n;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int bt(int l,int r)
{
    int x=++tot;
    if(l!=r)
    {
        int mid=(l+r)/2;
        tr[x].lc=bt(l,mid);
        tr[x].rc=bt(mid+1,r);
        tr[x].c=tr[tr[x].lc].c+tr[tr[x].rc].c;
    }
    else tr[x].c=1;
    return x;
}
void change(int x,int l,int r,int k,int c)
{
    if(l==r){tr[x].c=c;return;}
    int mid=(l+r)/2;
    if(k<=mid) change(tr[x].lc,l,mid,k,c);
    else change(tr[x].rc,mid+1,r,k,c);
    tr[x].c=tr[tr[x].lc].c+tr[tr[x].rc].c;
}
int find(int k)
{
    int x=1,l=1,r=n;
    while(l<r)
    {
        int mid=(l+r)/2;
        if(k<=tr[tr[x].lc].c) r=mid,x=tr[x].lc;
        else l=mid+1,k-=tr[tr[x].lc].c,x=tr[x].rc;
    }
    return l;
}
int findans(int x,int l,int r,int fl,int fr)
{
    if(l==fl&&fr==r) return tr[x].c;
    int mid=(l+r)/2;
    if(fr<=mid) return findans(tr[x].lc,l,mid,fl,fr);
    if(fl>mid) return findans(tr[x].rc,mid+1,r,fl,fr);
    return findans(tr[x].lc,l,mid,fl,mid)+findans(tr[x].rc,mid+1,r,mid+1,fr);
}
int main()
{
    n=read();
    bt(1,n);
    int last=1;
    for(int i=n;i>=1;i--)
    {
        int k;k=read();
        k%=i;int tmp=last+k;
        while(tmp>i) tmp-=i;
        last=find(tmp);
        change(1,1,n,last,0);
        printf("%d\n",last);
        last=findans(1,1,n,1,last)+1;
        if(last==i) last=1;
    }
}