結果

問題 No.1428 PeRmutation Question
ユーザー 箱星
提出日時 2020-12-27 20:43:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 33 ms / 2,000 ms
コード長 1,757 bytes
コンパイル時間 1,022 ms
コンパイル使用メモリ 87,572 KB
最終ジャッジ日時 2025-01-17 07:46:05
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
using ll = long long;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); i++)

const ll MOD = 1000000007;
ll modpow(ll a, ll n, ll mod) {
    ll res = 1;
    while (n > 0) {
        if (n & 1) res = res * a % mod;
        a = a * a % mod;
        n >>= 1;
    }
    return res;
}

ll modinv(ll i, ll mod) {
    return modpow(i, mod - 2, mod);
}

ll fact(ll n, ll mod) {
    ll ans = 1;
    rep(i, n) {
        ans = ans * (i + 1) % mod;
    }
    return ans;
}

ll comb(ll n, ll a, ll mod) {
    ll ans = 1, denom = 1;
    rep(i, a) {
        ans = ans * (n - i) % mod;
        denom = denom * (i + 1) % mod;
    }
    ans = ans * modinv(denom, mod) % mod;
    return ans;
}

int main()
{
    int n;
    cin >> n;
    vector<int> p(n);
    for (size_t i = 0; i < n; i++)
    {
        cin >> p[i];
        p[i]--;
    }
    map<int, int> mp;
    vector<bool> visited(n, false);
    for (size_t i = 0; i < n; i++)
    {
        if (!visited[i])
        {
            int l = 0;
            int v = i;
            while (!visited[v])
            {
                visited[v] = true;
                v = p[v];
                l++;
            }
            mp[l]++;
        }
    }
    bool half = true;
    ll ans = fact(n, MOD);
    for (auto itr = mp.begin(); itr != mp.end(); ++itr) {
        int k = itr->first;
        int v = itr->second;
        if (k % 2 == 0 || v >= 2)
        {
            half = false;
        }
        ans *= modinv(modpow(k, v, MOD), MOD);
        ans %= MOD;
        ans *= modinv(fact(v, MOD), MOD);
        ans %= MOD;
    }
    if (half && n != 1)
    {
        ans *= modinv(2, MOD);
        ans %= MOD;
    }
    cout << ans << "\n";
}
0