洛谷约瑟夫问题 (树状数组解法)

题目：约瑟夫问题 - 洛谷

题目描述

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

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

输入格式

输入两个整数 n,m。

输出格式

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

输入输出样例

输入 #1

10 3

输出 #1

3 6 9 2 7 1 8 5 10 4

说明/提示

1≤m,n≤100

#include<bits/stdc++.h>
using namespace std;
#define lowbit(x) ((-x)&(x))
typedef long long ll;
const int inf = 100010;
int n , m ;
int tree[inf] , arr[inf];

void add(int i , int k)
{
    for(;i<=n;i+=lowbit(i))
        tree[i]+=k;
}

int sum(int i)
{
    int ans = 0;
    for(;i>0;i-=lowbit(i))
        ans+=tree[i];
    return ans;
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        add(i,1);
    int tmp=n,i=0;
    while(tmp>1)
    {
        for(int k=1;k<=n;k++)
            cout<<tree[k]<<" ";
        cout<<endl;
        int j=(i+m-1)%tmp;//j为下面的求值做准备
        int p=j+1;//p表示这次出去的是剩余数组中的第几个元素 设该元素为X
        int l=1, r = n ;
        while(l<=r){//二分法在整个数组中查找X的位置
            int mid = (l+r)>>1;
            if(sum(mid)>=p) r=mid-1; //利用前缀和表示前面有几个元素 本题中树状数组的原数组每个元素为1，出列则为0
            else l=mid+1;
        }
        printf("j=%d p=%d tmp=%d l=%d r=%d\n",j,p,tmp,l,r);
        cout<<r+1<<endl;
        add(r+1,-1);//出列
        i=j%(--tmp);//数组元素数量减一 因为是圈，求出刚出列的元素的前一个元素在现在的数组中的位置
    }
    for(int k=1;k<=n;k++)
        cout<<tree[k]<<" ";
    cout<<endl;
    int l=1 , r=n;
    while(l<=r){
        int mid=(l+r)>>1;
        if(sum(mid)>=1) r=mid-1;
        else l=mid+1;
    }
    cout<<r+1<<endl;
    add(r+1,-1);
    for(int k=1;k<=n;k++)
        cout<<tree[k]<<" ";
    cout<<endl;
    return 0;
}

参考：NOJ1190约瑟夫问题线段树OR树状数组 - 百度文库 (baidu.com)

 有个貌似是更好的解法，但是我真没看懂题解 P1996 【约瑟夫问题】 - ysj1173886760 的博客 - 洛谷博客