結果

問題 No.1428 PeRmutation Question
コンテスト
ユーザー mine691
提出日時 2021-03-14 13:46:09
言語 C++17(gcc12)
(gcc 12.4.0 + boost 1.89.0)
コンパイル:
g++-12 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 2,159 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 5,682 ms
コンパイル使用メモリ 262,012 KB
実行使用メモリ 8,692 KB
最終ジャッジ日時 2026-06-18 18:35:27
合計ジャッジ時間 7,819 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <cassert>
#include "testlib.h"
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 argc, char* argv[]) {
    registerValidation(argc, argv);
    int n = inf.readInt(1, 100000, "n");
    inf.readEoln();
    vector<int> p(n);
    vector<int> counter(n, 0);
    for (size_t i = 0; i < n; i++)
    {
        p[i] = inf.readInt(1, n, "p_i");
        if (i != n - 1)
        {
            inf.readSpace();
        }
        p[i]--;
        counter[p[i]]++;
    }
    inf.readEoln();
    inf.readEof();
    for (size_t i = 0; i < n; i++)
    {
        assert(counter[i] == 1);
    }
    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