結果

問題 No.1978 Permutation Repetition
ユーザー xyz2606xyz2606
提出日時 2022-06-10 23:46:32
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 32 ms / 2,000 ms
コード長 5,258 bytes
コンパイル時間 2,827 ms
コンパイル使用メモリ 159,032 KB
実行使用メモリ 8,192 KB
最終ジャッジ日時 2024-09-21 07:01:55
合計ジャッジ時間 3,063 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 5 ms
7,836 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 5 ms
8,192 KB
testcase_09 AC 5 ms
7,204 KB
testcase_10 AC 5 ms
7,296 KB
testcase_11 AC 4 ms
6,912 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 3 ms
5,504 KB
testcase_14 AC 5 ms
7,840 KB
testcase_15 AC 4 ms
6,400 KB
testcase_16 AC 3 ms
5,376 KB
testcase_17 AC 3 ms
5,376 KB
testcase_18 AC 4 ms
7,680 KB
testcase_19 AC 3 ms
6,016 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 3 ms
5,376 KB
testcase_22 AC 4 ms
7,168 KB
testcase_23 AC 3 ms
6,272 KB
testcase_24 AC 3 ms
6,016 KB
testcase_25 AC 3 ms
6,528 KB
testcase_26 AC 3 ms
6,016 KB
testcase_27 AC 5 ms
7,040 KB
testcase_28 AC 5 ms
7,424 KB
testcase_29 AC 3 ms
6,784 KB
testcase_30 AC 4 ms
7,424 KB
testcase_31 AC 3 ms
6,144 KB
testcase_32 AC 3 ms
6,784 KB
testcase_33 AC 4 ms
6,656 KB
testcase_34 AC 3 ms
5,376 KB
testcase_35 AC 4 ms
6,912 KB
testcase_36 AC 3 ms
5,760 KB
testcase_37 AC 3 ms
5,376 KB
testcase_38 AC 2 ms
5,376 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 2 ms
5,376 KB
testcase_41 AC 4 ms
7,552 KB
testcase_42 AC 32 ms
8,020 KB
testcase_43 AC 20 ms
8,192 KB
testcase_44 AC 2 ms
5,376 KB
testcase_45 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<vector>
#include<set>
#include<map>
#include<queue>
#include<string>
#include<algorithm>
#include<iostream>
#include<bitset>
#include<functional>
#include<chrono>
#include<numeric>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cassert>
#include<cmath>
#include<iomanip>
#include<random>
#include<ctime>
#include<complex>
#include<type_traits>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;
typedef long long LL;
typedef double D;
#define all(v) (v).begin(), (v).end()
mt19937 gene(chrono::system_clock::now().time_since_epoch().count());
typedef complex<double> Complex;
#define fi first
#define se second
#define ins insert
#define pb push_back
inline char GET_CHAR(){
    const int maxn = 131072;
    static char buf[maxn],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,maxn,stdin),p1==p2)?EOF:*p1++;
}
inline int getInt() {
    int res(0);
    char c = getchar();
    while(c < '0') c = getchar();
    while(c >= '0') {
        res = res * 10 + (c - '0');
        c = getchar();
    }
    return res;
}
inline LL fastpo(LL x, LL n, LL mod) {
    LL res(1);
    while(n) {
        if(n & 1) {
            res = res * (LL)x % mod;
        }
        x = x * (LL) x % mod;
        n /= 2;
    }
    return res;
}
template<LL mod> struct Num {
	LL a;
	Num operator + (const Num & b) { return Num{(a + b.a) % mod}; }
	Num operator - (const Num & b) { return Num{(a - b.a + mod) % mod}; }
	Num operator * (const Num & b) { return Num{a * b.a % mod}; }
	Num operator / (const Num & b) { return Num{a * fastpo(b.a, mod - 2, mod) % mod}; }
	void operator += (const Num & b) {if((a += b.a) >= mod) a -= mod;}
	void operator -= (const Num & b) {if((a -= b.a) < 0) a += mod;}
	void operator *= (const Num & b) { a = a * b.a % mod; }
	void operator /= (const Num & b) { a = a * fastpo(b.a, mod - 2, mod) % mod; }
	void operator = (const Num & b) { a = b.a; }
	void operator = (const LL & b) { a = b; }
};
template<LL mod> ostream & operator << (ostream & os, const Num<mod> & a) {
	os << a.a;
	return os;
}
LL gcd(LL a, LL b) { return b ? gcd(b, a % b) : a; }

inline string itoa(LL x, int width = 0) {
	string res;
	if(x == 0) res.push_back('0');
	while(x) {
		res.push_back('0' + x % 10);
		x /= 10;
	}
	while((int)res.size() < width) res.push_back('0');
	reverse(res.begin(), res.end());
	return res;
}
const int _B = 131072;
char buf[_B];
int _bl = 0;
inline void flush() {
	fwrite(buf, 1, _bl, stdout);
	_bl = 0;
}
__inline void _putchar(char c) {
	if(_bl == _B) flush();
	buf[_bl++] = c;
}
inline void print(LL x, char c) {
	static char tmp[20];
	int l = 0;
	if(!x) tmp[l++] = '0';
	else {
		while(x) {
			tmp[l++] = x % 10 + '0';
			x /= 10;
		}
	}
	for(int i = l - 1; i >= 0; i--) _putchar(tmp[i]);
	_putchar(c);
}
typedef double C;
struct P {
	C x, y;
	void scan() {
		double _x, _y;
		scanf("%lf%lf", &_x, &_y);
		x = _x; y = _y;
	}
	void print() {
		cout << '(' << x << ' ' << y << ')' << endl;
	}
	P operator + (const P & b) const { return P{x + b.x, y + b.y}; }
	P operator - (const P & b) const { return P{x - b.x, y - b.y}; }
	C operator * (const P & b) const { return x * b.y - y * b.x; }
	C operator % (const P & b) const { return x * b.x + y * b.y; }
};
P operator * (const C & x, const P & b) { return P{x * b.x, x * b.y}; }
const int N = 1111;
const int LOG = 20;
const int mod = 1e9 + 7;
const int inf = 1e9 + 7;
int n, m;
int dx[4] = {1, 0, -1, 0};
int dy[4] = {0, 1, 0, -1};
int rela[N];
int getr(int x) {
    int p = x;
    while(rela[p] != p) p = rela[p];
    int p1 = p; p = x;
    while(rela[p] != p) {
        int p2 = rela[p];
        rela[p] = p1;
        p = p2;
    }
    return p1;
}
int c[N][N], fac[N], invfac[N], vst[N];
int a[N], dp[N], pd[N];
int main() {
	int n, m;
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
	}
	map<int, int> mp;
	vector<vector<int> > tot;
	for(int i = 1; i <= n; i++) {
		if(vst[i]) continue;
		int x = i;
		vector<int> vec;
		do {
			vec.pb(x);
			vst[x] = true;
			x = a[x];
		}while(x != i);
		tot.pb(vec);
		mp[vec.size()]++;
	}
	for(int i = 0; i <= n; i++) {
		for(int j = 0; j <= i; j++) {
			c[i][j] = !i || !j ? 1 : (c[i - 1][j - 1] + c[i - 1][j]) % mod;
		}
		fac[i] = i == 0 ? 1 : fac[i - 1] * (LL)i % mod;
		invfac[i] = fastpo(fac[i], mod - 2, mod);
	}
	LL ans = 1;
	for(auto t : mp) {
		vector<int> p;
		int k = t.fi;
		int num = t.se;
		for(int i = 1; i <= num; i++) {
			dp[i] = 0;
		}
		dp[0] = 1;

		for(int i = 1; i <= num; i++) {
			if(k * i / gcd(k * i, m) == k) {
				int w = fastpo(k, i - 1, mod) * fac[i - 1] % mod;
				//cout << i << ' ' << w << endl;
				memset(pd, 0, sizeof(int) * (num + 1));
				for(int j = 0; j <= num; j++) {
					LL cur = 1;
					for(int l = 0; j + l * i <= num; l++) {
						//cout << i << ' ' << j << ' ' << cur << endl;
						pd[j + l * i] = (pd[j + l * i] + dp[j] * cur % mod * invfac[l]) % mod;
						cur = cur * w % mod;
						cur = cur * c[num - j - l * i][i] % mod;
					}
				}
				memcpy(dp, pd, sizeof(int) * (num + 1));
			}
		}
		ans = (LL)ans * dp[num] % mod;
		
	}
	cout << ans << endl;

}
0