PKU 3641 Pseudoprime numbers 快速模取幂素数

最新推荐文章于 2013-04-11 21:05:23 发布

原创最新推荐文章于 2013-04-11 21:05:23 发布 · 854 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#numbers #user #c

【ACM】专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一个使用C++实现的快速模取幂算法和素数判断方法，并通过一个具体的问题实例展示了这两种算法的应用。该程序能够根据输入的数值p和m，计算m的p次方模p的结果，并判断p是否为素数。

/*
----------------------------------
    
    Problem: 3641		User: a312745658
    Memory: 192K		Time: 16MS
    Language: C++		Result: Accepted
    
    stratege: 快速模取幂 + 判断素数
    Author ： Johnsondu
-----------------------------------

*/

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

long long res ;
long long m, p ;

bool isPrime (int n)
{
    int i ;
    for (i = 2; i * i <= n; i ++)
        if (n % i == 0)
            return true ;
    return false ;
}

long long exp_mod (long long a, long long n, long long b)
{
    long long t = 1;

    while (n)
    {
        if (n & 1)
            t = (t * a) % b ;
        a = (a * a) % b ;
        n >>= 1 ;
    }

    return t % b;
}

int main()
{
    while (cin >> p >> m, p + m)
    {
        res = exp_mod (m, p, p) ;
        if (res == m%p && isPrime (p))
            printf ("yes\n") ;
        else
            printf ("no\n") ;
    }
    return 0 ;
}