找规律-POJ3372

http://poj.org/problem?id=3372

一群小孩围成一个圈，老师顺时针发糖，分别每间隔0，1, 2, 3, 4……发一颗，问是否每个同学都有糖

打印了前几项看出来得规律，只要n是2^k则输出YES，否则输出NO

数论解法：

N个人围成一个圈,老师给N个人分糖果.第i次分到的人的编号是f(x) = (x*(x+1)/2)%N,这N个人能不能至少获得1个糖果.

易知必然存在i使得f(N - i - 1) = f(N + i) (mod N), 即周期为N,

只要判断f(x)能否构成N的完全剩余系，就能得出结果

f(x)为N的完全系,即证不存在i,j(i != j),使得f(i) == f(j)

            i*(i - 1)/2 = j*(j-1)/2 (mod N)

      (i-j)*(i+j+1)/2 = 0 (mod N)

其中(i-j)与(i+j+1)异号.

假设(i-j)为偶数

若(i-j)/2为偶数,当N为S*2^l(S奇数,l > 0),有解

若(i-j)/2为奇数,当N为P*Q,(P,Q是奇数),有解

类似当(i+j+1)为偶数时也一样.

因此N只能是2^l(l > 0)

另外学会得东西是如何判断n是否为2^k

n&(n-1) == 0 true

因为2得k次方得二进制中一定只有一个1

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#define ll long long
using namespace std;

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        if(n&(n-1))
            printf("NO\n");
        else
            printf("YES\n");
    }
    return 0;
}