[洛谷]P1996 约瑟夫问题 (#树状数组)

题目背景

约瑟夫是一个无聊的人！！！

题目描述

n个人(n<=100)围成一圈,从第一个人开始报数,数到m的人出列,再由下一个人重新从1开始报数,数到m的人再出圈,……依次类推,直到所有的人都出圈,请输出依次出圈人的编号.

输入输出格式

输入格式：

n m

输出格式：

出圈的编号

输入输出样例

输入样例#1

10 3

输出样例#1

3 6 9 2 7 1 8 5 10 4

---

思路

树状数组写的。

#include <stdio.h>
#include <iostream>
using namespace std;
int n,m,bit[100001],maxn,s;
inline int lowbit(int x)
{
	return x&(-x);
}
inline void del(int pos,int x)
{
	register int i;
	for(i=pos;i<=maxn;i+=lowbit(i))
	{
		bit[i]+=x;
	}
}
inline int find(int i)
{
	register int j,ans(0),now(0);
	for(j=30;j>=0;j--)
	{
		ans+=(1<<j);
		if(ans>maxn || bit[ans]+now>=i)
		{
			ans-=(1<<j);
		}
		else
		{
			now+=bit[ans];
		}
	}
	return ans+1;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i;
	cin>>n>>m;
	maxn=n;
	for(i=1;i<=n;i++)
	{
		bit[i]=lowbit(i);
	}
	int now(1);
	while(n)
	{
		now=(now-1+m-1)%n+1;
		s=find(now);
		del(s,-1);
		cout<<s<<' ';
		n--;
	}
	cout<<endl;
	return 0;
}