Codeforces Round #334 (Div. 2) D. Modular Arithmetic(置换)

最新推荐文章于 2022-11-22 23:15:14 发布

ACMiao_

最新推荐文章于 2022-11-22 23:15:14 发布

阅读量377

点赞数

分类专栏：置换群基础数学文章标签：置换群

本文链接：https://blog.csdn.net/Yeluorag/article/details/50151797

版权

基础数学同时被 2 个专栏收录

16 篇文章 0 订阅

订阅专栏

置换群

5 篇文章 0 订阅

订阅专栏

题意：给出方程 f(kx%p)=kf(x)%p ，问在集合A->B上不同的映射函数f有几种，其中A=B={0,1,2..p-1}，p为素数（除了2），k为小于p的一个常数。

思路：f(kx1%p) = kf(x0)%p,

f(kx2%p) = kf(x1)%p = k^2 * f(x0)%p,

f(kx3%p) = ... = k^3 * f(x0)%p,

......

f(kxm%p) = ... = k^m * f(x0)%p.

显然，由于gcd(k, p) == 1，由费马小定理最终一定存在m使得k^m % p == 1，且m为p-1的因子。这样就形成了环。对每个环，起始位置有p种可能，所以 ans = p ^ cnt。特判k为0和1的情况。

#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <cmath>
#include <set>
#include <map>
using namespace std;

typedef long long LL;
#define mem(a, n) memset(a, n, sizeof(a))
#define ALL(v) v.begin(), v.end()
#define si(a) scanf("%d", &a)
#define sii(a, b) scanf("%d%d", &a, &b)
#define siii(a, b, c) scanf("%d%d%d", &a, &b, &c)
#define pb push_back
#define eps 1e-8
const int inf = 0x3f3f3f3f, N = 1e6 + 5, MOD = 1e9 + 7;

int T, cas = 0;
bool vis[N];
LL p, k;
LL quick_mod(LL a, LL b, LL m) {
    LL ret = 1;
    a %= m;
    while(b > 0) {
        if(b & 1) ret = ret * a % m;
        a = a * a % m;
        b >>= 1;
    }
    return ret;
}

int main(){
#ifdef LOCAL
    freopen("/Users/apple/input.txt", "r", stdin);
//	freopen("/Users/apple/out.txt", "w", stdout);
#endif
    
    cin >> p >> k;
    if(k == 0) {
        cout << quick_mod(p, p - 1, MOD) << endl;
        return 0;
    } if(k == 1) {
        cout << quick_mod(p, p, MOD) << endl;
        return 0;
    }
    mem(vis, false);
    LL ans = 1; int cnt = 0;
    for(int i = 0; i < p; i ++) {
        int idx = i, flag = 0;
        while(!vis[idx]) {
            vis[idx] = 1; flag ++;
            idx = 1LL * idx * k % p;
        }
        cnt += flag > 1;
    }
    cout << quick_mod(p, cnt, MOD) << endl;
    return 0;
}