Intelligence math problem || 判断一个数是否为2的k次方

Description

 

一共有N个孩子，按顺时针方向从1到N排成一圈的孩子们正在等糖果。他们的老师按以下方式分发糖果：首先，老师给1号和2号孩子每人一个糖果。然后他顺时针沿着圆圈走，跳过一个孩子（3号孩子），给下一个孩子（4号孩子）一个糖果。然后他继续散步，跳过两个孩子（5号和6号孩子），给下一个孩子（7号孩子）一块糖果，然后跳过三个孩子在发一个糖果等等。现在你必须告诉老师在无限走下去的情况下所有的孩子是否都会得到至少一个糖果？

Input

 

第一行包含一个整数 T，接下来有T组测试数据，对于每一组测试数据，输入一个整数n Data limit:1≤T≤60,1≤n≤10^18

Output

 

对于每一组测试数据，输出对应答案.如果能分到请输出YES，否则输出NO

Sample Input 1 

2
2
3

Sample Output 1

YES
NO

 

分析：拿到这道题的时候，第一反应想到约瑟夫环，但是好像又和约瑟夫环并不一样，特别是看到教师可以一直走下这个条件时，如果写让教师一直走下去的代码，只有当所有学生有糖果时循环才会截止，如果遇到并不是所有孩子都能得到糖的时候，代码将一直跑下去。这时我猜想一定有规律所寻或者是存在一个公式。首先我采用了打表，假设教师只走1000步，从一个孩子开始看哪些孩子能够得到糖果，也就是我注释掉的地方，经过多个孩子我发现，教师给的孩子是有规律的，是2n的循环。举个例子，如果是两个孩子   教师给糖果时（阿拉伯数字表示给第几个孩子，我从0开始）1 0 0 1 1 0 0 1 1 0 0 1 （发现是2n的循环，n=2），如果是3个孩子：1 2 1 1 2 1 1 2 1 1 2 1，如果是4个孩子：1 2 0 3 3 0 2 1

这时我将孩子们从小到大排序，记录不同孩子的数量，如果f<y,则不是所有的孩子都有糖果。

下面是我的打表代码：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

int a[200005];

int main(){
    int n;
    n=1000;
    int y;
    while(scanf("%d",&y)!=EOF){
    int f=1;
    int k=0;
    int i=0;
    long long r=1;
    int t=1000;
    while(n--){
        r=r+k;
        a[i]=r%y;
        //cout<<a[i]<<" ";//判断哪些孩子得到糖 
        k++;
        i++;
    }
    //cout<<endl;
    sort(a,a+t);
    //cout<<a[0]<<" ";
    for(int m=1;m<t;m++){
        if(a[m]!=a[m-1]){
            f++;
            //cout<<a[m]<<" ";
        }
    }
    //cout<<endl;
    //cout<<"f: "<<f<<endl;
    if(f<y){
        cout<<"NO"<<endl;

    }
    else{
        cout<<"YES"<<endl;
    }
    n=1000;
    cout<<endl;
    }
    
    return 0;
} 

 

打表分析：通过打表，我发现，只有孩子的数量是2的k次方时，所有孩子才能分得糖果，下面是用判断一个数是否为2的K次方的模板写的

 

代码：

#include <iostream>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>

using namespace std;

typedef long long int LL;

set<LL> tab;

void Init() {
    LL tmp = 1;
    tab.insert(1);
    for (int i = 0; i < 60; i++) {
        tmp = tmp << 1;
        tab.insert(tmp);
    }
    return;
}

// 检查num这个数是否是2的k次方
int check(LL num) {
    if (tab.count(num) == 1)
        return 1;
    else
        return 0;
}

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    Init();
    int t;
    cin>>t;
    while(t--){
         int n;
    cin>>n;
    int sum=check(n); 
    if(sum==1){
        cout<<"YES"<<endl;
    }
    else{
        cout<<"NO"<<endl;
    }
    }
   
    return 0;
}