ural1521-線段樹解約瑟夫問題

使用sum[]數組來記錄當前區段的個數

pushup來更新子節點與父節點的關係，實現很簡單，但是要注意每次節點個數在減少，注意查詢時要求的長度要模一下當前節點的個數

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn=100010;

int sum[maxn<<2];


void pushup(int  idx)
{
	sum[idx]=sum[idx<<1]+sum[idx<<1|1];
}

void build(int left,int right,int idx)
{
	sum[idx]=right-left+1;
	if(right==left)
		return ;
	int mid=(right+left)>>1;
	build(left,mid,idx<<1);
	build(mid+1,right,idx<<1|1);
}

int query(int val,int left,int right,int idx)
{
	if(right==left)
	{
		sum[idx]=0;
		return left;
	}
	int mid=(right+left)>>1;
	int ans;
	if(sum[idx<<1]>=val)
		ans=query(val,left,mid,idx<<1);
	else
		ans=query(val-sum[idx<<1],mid+1,right,idx<<1|1);
	pushup(idx);
	return ans;
}

int main()
{
	int N,K;

	scanf("%d%d",&N,&K);
	build(1,N,1);
	int temp=K;
	for(int i=1;i<=N;i++)
	{
		if(i!=N)
		printf("%d ",query(temp,1,N,1));
		else
			printf("%d",query(temp,1,N,1));
		if(i!=N)
			temp=(temp+K-1)%(N-i);
		if(temp==0)
			temp=N-i;
	}

}