码迷,mamicode.com
首页 > 其他好文 > 详细

【模板】质数判断(Miller_Rabin)

时间:2018-08-06 12:58:51      阅读:111      评论:0      收藏:0      [点我收藏+]

标签:code   ace   cst   amp   break   pre   lin   clu   std   

题意简述

给定一个范围N,你需要处理M个某数字是否为质数的询问(每个数字均在范围1-N内)

题解思路

费马小定理: n是一个奇素数,a是任何整数(\(1≤ a≤n-1\)) ,则\(a^{p-1}≡1(mod\ p)\)
推论:如果n是一个奇素数,则方程\(x^2 ≡ 1 (mod\ n)\)只有±1两个解

代码

#include <cstdio>
using namespace std;
const int t[5] = {0, 2, 7, 61};
int n, m, x;
int ksm(int a, int r, int mod)
{
    if (r == 0)
        return 1;
    if (r == 1)
        return a;
    int x = ksm(a, r >> 1, mod) % mod;
    if (r & 1)
        return ((long long) x * x * a) % mod; 
    else return ((long long) x * x) % mod;
}
bool mr(int x)
{
    if (x == 1) 
        return 0;
    int cnt = 0, p1 = x - 1;
    while (p1 % 2 == 0)
    {
        ++cnt;
        p1 /= 2;
    }
    for (int i = 1; i <= 3; ++i)
    {
        if (x == t[i])
            return 1;
        int xx = ksm(t[i], p1, x);
        if (xx % x != 1 && xx % x != x - 1)
        {
            bool flag = 0;
            for (int j = 1; j <= cnt; ++j)
            {
                xx = (long long) xx * xx % x;
                if (xx == x - 1)
                {
                    flag = 1; 
                    break;
                }
            }
            if (!flag)
                return 0;
        } 
    }
    return 1;
}
int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= m; ++i)
    {
        scanf("%d", &x);
        if (mr(x)) puts("Yes");
        else puts("No");
    }
}

【模板】质数判断(Miller_Rabin)

标签:code   ace   cst   amp   break   pre   lin   clu   std   

原文地址:https://www.cnblogs.com/xuyixuan/p/9428203.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!