1521. War Games 2(线段树解约瑟夫）

1521

根据区间和 来确定第k个数在哪

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define N 100010
int s[N<<2],tt;
void up(int w)
{
    s[w] = s[w<<1]+s[w<<1|1];
}
void build(int l,int r,int w)
{
    if(l==r)
    {
        s[w] = 1;
        return ;
    }
    int m = (l+r)>>1;
    build(l,m,w<<1);
    build(m+1,r,w<<1|1);
    up(w);
}
void update(int k,int l,int r,int w)
{
    if(l==r)
    {
        s[w] = 0;
        tt = l;
        return ;
    }
    int m = (l+r)>>1;
    if(k<=s[w<<1])
    update(k,l,m,w<<1);
    else
    update(k-s[w<<1],m+1,r,w<<1|1);
    up(w);
}
int getsum(int a,int b,int l,int r,int w)
{
    if(a<=l&&b>=r)
    {
        return s[w];
    }
    int m = (l+r)>>1,re=0;
    if(a<=m)
    re+=getsum(a,b,l,m,w<<1);
    if(b>m)
    re+=getsum(a,b,m+1,r,w<<1|1);
    return re;
}
int main()
{
    int k,n,o=0,ss=0;
    scanf("%d%d",&n,&k);
    build(1,n,1);
    int m = n;
    while(n)
    {
        if(tt)
        {
            ss = getsum(1,tt,1,m,1);
            o = (k+ss)%s[1];
        }
        else
        o = k%s[1];
        if(o==0)
        o=s[1];
        update(o,1,m,1);
        n--;
        if(n)
        printf("%d ",tt);
        else
        printf("%d
",tt);
    }
    return 0;
}