HDU 4861 （找规律）

最新推荐文章于 2018-10-11 23:25:20 发布

xionghao-coder

最新推荐文章于 2018-10-11 23:25:20 发布

阅读量159

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM

本文链接：https://blog.youkuaiyun.com/weixin_38378637/article/details/81146692

ACM 专栏收录该内容

69 篇文章

订阅专栏

本文探讨了给定k个数字val(i)=1^i+2^i+...+(p-1)^i(mod p)的问题，其中p为素数。通过对数值规律的观察，文章揭示了当且仅当i等于p-1时，val(i)才等于p-1，其余情况下val(i)均为0。基于此规律，提出了一种快速判断方法。

题意：
有k个数字，第i个值val(i)=1^i+2^i+...+(p-1)^i (mod p) (1<=i<=k)，p为素数。

思路：
找规律，发现对于质数，只有i=p-1时才有val(i)=p-1，其他情况val(i)=0。

代码：

#include <iostream>
#include <iomanip>
#include <algorithm>
#include <cstring>
#include <cctype>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <map>
#include <list>
#include <set>
#include <stack>
#include <queue>
#include <string>
#include <sstream>
#define pb push_back
#define X first
#define Y second
#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x,x.begin())
#define pii pair<int,int>
#define qclear(a) while(!a.empty())a.pop();
#define lowbit(x) (x&-x)
#define sd(n) scanf("%d",&n)
#define sdd(n,m) scanf("%d%d",&n,&m)
#define sddd(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define mst(a,b) memset(a,b,sizeof(a))
#define cout3(x,y,z) cout<<x<<" "<<y<<" "<<z<<endl
#define cout2(x,y) cout<<x<<" "<<y<<endl
#define cout1(x) cout<<x<<endl
#define IOS std::ios::sync_with_stdio(false)
#define SRAND srand((unsigned int)(time(0)))
typedef long long ll;
typedef unsigned long long ull;
typedef unsigned int uint;
using namespace std;
const double PI=acos(-1.0);
const int INF=0x3f3f3f3f;
const ll INFF=0x3f3f3f3f3f3f3f3f;
const ll mod=998244353;
const double eps=1e-5;
const int maxn=1005;
const int maxm=20005;


void solve() {
    int k,p;
    while(~sdd(k,p)){
        if((k/(p-1))&1){
            printf("YES\n");
        }else{
            printf("NO\n");
        }
    }
    return ;
}
int main() {
#ifdef LOCAL
    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
#else
    //    freopen("","r",stdin);
    //    freopen("","w",stdout);
#endif
    solve();
    return 0;
}