HDU 5860 Death Sequence

用线段树可以算出序列。然后o(1)询问。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0),eps=1e-;
void File()
{
    freopen("D:\\in.txt","r",stdin);
    freopen("D:\\out.txt","w",stdout);
}
template <class T>
inline void read(T &x)
{
    char c = getchar(); x = ;while(!isdigit(c)) c = getchar();
    while(isdigit(c)) { x = x *  + c - ''; c = getchar();  }
}

const int maxn=;
int T,n,k,q,sz,cnt,p;
int s[*maxn],ans[maxn];

void build(int l,int r,int rt)
{
    s[rt]=;
    if(l==r){ s[rt]=; return;}
    int m=(l+r)/;
    build(l,m,*rt);
    build(m+,r,*rt+);
    s[rt]=s[*rt]+s[*rt+];
}

void get(int sum,int l,int r,int rt)
{
    if(l==r) { s[rt]=, p=l; return; }
    int m=(l+r)/;
    if(s[*rt]>=sum) get(sum,l,m,*rt);
    else get(sum-s[*rt],m+,r,*rt+);
    s[rt]=s[*rt]+s[*rt+];
}

int main()
{
   // File();
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&k,&q);
        build(,n,);

        sz=n; cnt=;
        while(sz>)
        {
            int mx=(sz-)/k;
            for(int i=;i<=mx;i++)
            {
                int sum=+i*k-i;
                get(sum,,n,); ans[++cnt]=p; sz--;
            }
        }
        for(int i=;i<=q;i++)
        {
            int x; scanf("%d",&x);
            printf("%d\n",ans[x]);
        }
    }
    return ;
}