题意:桌上有k个小球,第i个小球的值value(i) = 1^i + 2^i + 3^i + ... + (p-1)^i (mod p),其中p为素数。两个人轮流取球,胜者为取到所有球总和较大的人。求问:先手能否取胜?
分析一:
费马小定理:a^(p-1) = 1 ( mod p) , 其中a与p互质。
因此,当i = p-1 时,value(i) = p-1;
分析二:
假设a是关于p的原根(原根定义参见百度),那么存在:
(a^1)^i, (a^2)^i, (a^3)^i, ... , (a^(p-1))^i 数列对于p取模后的结果,必然和 1^i, 2^i, 3^i, ... (p-1)^i 数列取模后,是同一个排列。
因此,value(i) % p = (a^1)^i + (a^2)^i + (a^3)^i + ... + (a^(p-1))^i = ((a^i)^p - a^i) / (a^i - 1),易证: value(i)%p = 0 .
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
int main(){
int k,p;
while(scanf("%d%d",&k,&p)!=EOF){
if((k/(p-1))&1) printf("YES\n");
else printf("NO\n");
}
return 0;
}