P1996 约瑟夫问题 （树状数组）

题目描述

nn 个人围成一圈，从第一个人开始报数,数到 mm 的人出列，再由下一个人重新从 11 开始报数，数到 mm 的人再出圈，依次类推，直到所有的人都出圈，请输出依次出圈人的编号。

注意：本题和《深入浅出-基础篇》上例题的表述稍有不同。书上表述是给出淘汰 n-1n−1 名小朋友，而该题是全部出圈。

输入格式

输入两个整数 n,mn,m。

输出格式

输出一行 nn 个整数，按顺序输出每个出圈人的编号。

输入输出样例

输入 #1复制

10 3

输出 #1复制

3 6 9 2 7 1 8 5 10 4

 思路：

其实就是存在的人从1喊到m，存在且喊到m的人出局

假设下标从1开始，从now开始喊，那么下一个喊到的人应该是now+m-1

但是由于喊出的人要出局，当前人数会有变化，假设当前人数为op，那么我们就得将now映射到1~op序号里

那么我们先将now+m-1减去1，将他映射到0~n-1的序号里，再%op将他映射到0~op-1的序号里，最后加上1将他映射到1~op的序号里，即下次喊到m的人的序号x是：x=(now+m-1-1)%op+1

上面我们算的是当人都存在的时候，喊到的人的序号，那么随着出局会有人不存在，那么我们就用树状数组来记录每个人存在与否，每个人的位置刚开始都为1表示都存在，当他喊到的时候将他位置上的数-1，并且将记录当前人数的op--。由于x算的是从存在的人里算出的序号，那么我们最终算的人的真实序号id其实是满足sum(id）==x的id，id用二分求出，然后直接输出即可

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=200;
int n,m;
int a[N];
int tr[4*N];
int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i))res+=tr[i];
	return res;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i))tr[i]+=c;
}
bool cheek(int x,int s){
	if(sum(x)>=s)return true;
	else return false;
}
void sove(){
	cin>>n>>m;
	int now=1;
	int op=n;
	for(int i=1;i<=n;i++)add(i,1);
	while(op){
		now=(now+m-1-1)%op+1;
		int l=1,r=n;
		while(l<r){
			int mid=l+r>>1;
			if(cheek(mid,now))r=mid;
			else l=mid+1;
		}
		cout<<l<<" ";
		add(l,-1);
		op--;
	}
	
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(),cout.tie();
	int t=1;
//	cin>>t;
	while(t--){
		sove();
	}
	return 0;
}