CodeForces 603 B. Moodular Arithmetic（数论+并查集）

Description
给出一个整数k以及一个奇素数p，问满足f(kx%p)=k*f(x)%p的映射f：{0,1,…,p-1}->{0,1,…,p-1}的数量（结果模1e9+7）
Input
两个整数k和p(3 ≤ p ≤ 1 000 000, 0 ≤ k ≤ p - 1)
Output
输出满足条件的映射f的个数
Sample Input
3 2
Sample Output
3
Solution
当k=0时，f(0)=0，即除f(0)已经确定外其余p-1个函数值都可以任意选，所以此时答案是p^(p-1)
当k=1时，f(x)=f(x)，这个条件等同没有，所有f均满足条件，所以答案就是p^p
当k>=2时，任意取定一个x，那么每一个确定的f(x)都会确定f(kx%p)，那么这些x就将f(x)分成了几个集合，集合中的函数值总共有p种取法，所以问题转化为求这种集合的个数m或者叫统计环数，这个可以用vis数组通过标记来记录，这里我用的是并查集，那么答案就是p^m。
注意f(0)是被唯一确定的，所以不要考虑f(0)
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 1111111
typedef long long ll;
#define mod 1000000007ll
int k,p;
int par[maxn];//par[i]表示i的父亲的编号 
int rank[maxn];//rank[i]表示i所在树的高度 
void init(int n)//初始化n个元素 
{
    for(int i=0;i<n;i++)
    {
        par[i]=i;
        rank[i]=0;
    }
} 
int find(int x)//查询树的根 
{
    if(par[x]==x) return x;
    else return par[x]=find(par[x]);
}
void unite(int x,int y)//合并x和y所属的集合 
{
    x=find(x);
    y=find(y);
    if(x==y) return;
    if(rank[x]<rank[y]) par[x]=y;
    else
    { 
        par[y]=x;
        if(rank[x]==rank[y]) rank[x]++;
    } 
}
bool same(int x,int y)//判断x和y是否属于同一个集合 
{
    return find(x)==find(y);
}
ll mod_pow(ll a,ll b,ll p)//快速幂,计算a^b mod p 
{
    ll ans=1ll;
    a%=p;
    while(b)
    {
        if(b&1) ans=(ans*a)%p;
        a=(a*a)%p;
        b>>=1;
    } 
    return ans;
}
int main()
{
    while(~scanf("%d%d",&p,&k))
    {
        if(k==0)printf("%I64d\n",mod_pow(p,p-1,mod));//除了f(0)被确定之外其他值可以随便取 
        else if(k==1)printf("%I64d\n",mod_pow(p,p,mod));//所有值都可以随便取 
        else
        {
            init(p);
            //注意f(0)已经被确定 
            for(int i=1;i<p;i++)
            {
                int j=1ll*k*i%p;
                if(!same(i,j))unite(i,j); 
            }
            ll ans=1;
            for(int i=1;i<p;i++)//找集合的个数,每个集合p种不同的取法 
                if(i==par[i])
                    ans=ans*p%mod;
            printf("%I64d\n",ans);
        } 
    }
    return 0;
}