hdu 4992 Primitive Roots 【求原根模板】

题目链接

大题流程： 判定是否有原根->求出最小原根->利用最小原根找出全部原根

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;

const int maxn=1e6+;
int prime[maxn+];
bool check[maxn+];
int phi[maxn+];
int num_prime;
void init()
{
    memset(check, false, sizeof(check));
    phi[]=;
    for(int i=; i<=maxn; i++)
    {
        if(!check[i])
        {
            prime[num_prime++]=i;
            phi[i]=i-;
        }
        for(int j=; j<num_prime; j++)
        {
            if(i*prime[j]>maxn) break;
            check[i*prime[j]]=true;
            if(i%prime[j]==)
            {
                phi[i*prime[j]]=phi[i]*prime[j];
                break;
            }
            else
            {
                phi[i*prime[j]]=phi[i]*(prime[j]-);
            }
        }
    }
}

LL gcd(LL a, LL b)
{
    return b? gcd(b,a%b):a;
}

void get(LL n,vector<LL>& fac)    //对n进行因式分解
{
    fac.clear();
    for(LL i=; i*i<=n; i++)
        if(n%i==)
        {
            fac.push_back(i);
            while(n%i==) n/=i;
        }
    if(n>) fac.push_back(n);
}

LL qpow(LL x,LL n,LL mod)        //求x^n%mod
{
    LL ret=;
    for(; n; n>>=)
    {
        if(n&) ret=ret*x%mod;
        x=x*x%mod;
    }
    return ret;
}

vector<LL> fac;
vector<LL> ans;

bool ok(LL x)
{
    if(x%==) x/=;
    if(x%==) return false;
    for(int i=; prime[i]*prime[i]<=x; i++) if(x%prime[i]==)
        {
            while(x%prime[i]==) x/=prime[i];
            return x==;
        }
    return true;
}

LL get_g(LL p)        //得到一个正整数p的最小原根
{
    for(int i=; i<p; i++)
    {
        bool flag=false;
        for(LL x:fac)
            if(qpow(i,phi[p]/x,p)==)
            {
                flag=true;
                break;
            }
        if(!flag&&qpow(i,phi[p],p)==)  return i;
    }
}

void GetAns(LL g,LL p,vector<LL>& ans)    //由最小原根p，得到某正整数p的全部原根
{
    ans.clear();
    ans.push_back(g);
    for(int i=; i<phi[p]; i++)
        if(gcd(i,phi[p])==) ans.push_back(qpow(g,i,p));
}

int main()
{
    init();
    LL p;
    while(~scanf("%lld",&p))
    {
        if(p==||p==)
        {
            printf("%lld\n",p-);
            continue;
        }
        if(!ok(p))        //首先判断是否有原根
        {
            puts("-1");
            continue;
        }
        get(phi[p],fac);
        LL g=get_g(p);        //定义g为最小原根
        GetAns(g,p,ans);
        sort(ans.begin(),ans.end());
        for(int i=; i<ans.size(); i++)
            printf("%lld%c",ans[i],i==ans.size()-? '\n':' ');
    }
}