結果

問題 No.1428 PeRmutation Question
ユーザー mine691mine691
提出日時 2021-03-14 13:46:09
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 2,159 bytes
コンパイル時間 7,768 ms
コンパイル使用メモリ 248,332 KB
実行使用メモリ 8,672 KB
最終ジャッジ日時 2024-04-23 21:07:25
合計ジャッジ時間 10,451 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 6 ms
5,376 KB
testcase_13 AC 24 ms
6,392 KB
testcase_14 AC 23 ms
6,244 KB
testcase_15 AC 16 ms
6,096 KB
testcase_16 AC 30 ms
8,624 KB
testcase_17 AC 30 ms
8,480 KB
testcase_18 AC 12 ms
5,376 KB
testcase_19 AC 9 ms
5,376 KB
testcase_20 AC 10 ms
5,376 KB
testcase_21 AC 27 ms
6,408 KB
testcase_22 AC 32 ms
8,528 KB
testcase_23 AC 31 ms
8,540 KB
testcase_24 AC 32 ms
8,400 KB
testcase_25 AC 27 ms
6,400 KB
testcase_26 AC 27 ms
6,396 KB
testcase_27 AC 34 ms
8,540 KB
testcase_28 AC 27 ms
6,548 KB
testcase_29 AC 31 ms
8,532 KB
testcase_30 AC 31 ms
8,544 KB
testcase_31 AC 30 ms
8,672 KB
testcase_32 AC 30 ms
8,416 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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