結果

問題 No.1428 PeRmutation Question
ユーザー mine691mine691
提出日時 2021-03-14 13:46:09
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,159 bytes
コンパイル時間 2,918 ms
コンパイル使用メモリ 38,960 KB
最終ジャッジ日時 2025-01-19 16:28:14
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。(リジャッジ後のコンパイルエラーは公開されます)
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp:6:10: fatal error: testlib.h: No such file or directory
    6 | #include "testlib.h"
      |          ^~~~~~~~~~~
compilation terminated.

ソースコード

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