結果
| 問題 |
No.1392 Don't be together
|
| コンテスト | |
| ユーザー |
 fuppy_kyopro
|
| 提出日時 | 2021-02-12 22:32:37 |
| 言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
CE
(最新)
AC
(最初)
|
| 実行時間 | - |
| コード長 | 5,262 bytes |
| コンパイル時間 | 2,496 ms |
| コンパイル使用メモリ | 178,772 KB |
| 最終ジャッジ日時 | 2025-01-05 10:39:11 |
| 合計ジャッジ時間 | 3,075 ms |
|
ジャッジサーバーID （参考情報） |
judge4 / judge3 |
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
コンパイルメッセージ
In file included from /usr/include/c++/13/string:43,
from /usr/include/c++/13/bitset:52,
from /usr/include/x86_64-linux-gnu/c++/13/bits/stdc++.h:52,
from main.cpp:7:
/usr/include/c++/13/bits/allocator.h: In destructor ‘std::_Vector_base<int, std::allocator<int> >::_Vector_impl::~_Vector_impl()’:
/usr/include/c++/13/bits/allocator.h:184:7: error: inlining failed in call to ‘always_inline’ ‘std::allocator< <template-parameter-1-1> >::~allocator() noexcept [with _Tp = int]’: target specific option mismatch
184 | ~allocator() _GLIBCXX_NOTHROW { }
| ^
In file included from /usr/include/c++/13/vector:66,
from /usr/include/c++/13/queue:63,
from /usr/include/x86_64-linux-gnu/c++/13/bits/stdc++.h:157:
/usr/include/c++/13/bits/stl_vector.h:133:14: note: called from here
133 | struct _Vector_impl
| ^~~~~~~~~~~~
ソースコード
//*
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
//*/
#include <bits/stdc++.h>
// #include <atcoder/all>
using namespace std;
// using namespace atcoder;
#define DEBUG(x) cerr<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cerr<<#v<<":";for(int i=0;i<v.size();i++) cerr<<" "<<v[i]; cerr<<endl;
#define DEBUG_MAT(v) cerr<<#v<<endl;for(int i=0;i<v.size();i++){for(int j=0;j<v[i].size();j++) {cerr<<v[i][j]<<" ";}cerr<<endl;}
typedef long long ll;
// #define int ll
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
#define pll pair<ll,ll>
template<class S, class T> pair<S, T> operator+(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first + t.first, s.second + t.second); }
template<class S, class T> pair<S, T> operator-(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first - t.first, s.second - t.second); }
template<class S, class T> ostream& operator<<(ostream& os, pair<S, T> p) { os << "(" << p.first << ", " << p.second << ")"; return os; }
#define X first
#define Y second
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep1(i,n) for(int i=1;i<=(int)(n);i++)
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;i--)
#define rrep1(i,n) for(int i=(int)(n);i>0;i--)
#define REP(i,a,b) for(int i=a;i<b;i++)
#define in(x, a, b) (a <= x && x < b)
#define all(c) c.begin(),c.end()
void YES(bool t=true) {cout<<(t?"YES":"NO")<<endl;}
void Yes(bool t=true) {cout<<(t?"Yes":"No")<<endl;}
void yes(bool t=true) {cout<<(t?"yes":"no")<<endl;}
void NO(bool t=true) {cout<<(t?"NO":"YES")<<endl;}
void No(bool t=true) {cout<<(t?"No":"Yes")<<endl;}
void no(bool t=true) {cout<<(t?"no":"yes")<<endl;}
template<class T> bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a = b; return 1; } return 0; }
#define UNIQUE(v) v.erase(std::unique(v.begin(), v.end()), v.end());
const ll inf = 1000000001;
const ll INF = (ll)1e18 + 1;
const long double pi = 3.1415926535897932384626433832795028841971L;
int popcount(ll t) { return __builtin_popcountll(t); }
// int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};
// int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };
vi dx = {0, 1, 0, -1}, dy = {-1, 0, 1, 0};
// vi dx2 = { 1,1,0,-1,-1,-1,0,1 }, dy2 = { 0,1,1,1,0,-1,-1,-1 };
struct Setup_io {
Setup_io() {
ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
cout << fixed << setprecision(25);
}
} setup_io;
// const ll MOD = 1000000007;
const ll MOD = 998244353;
// #define mp make_pair
//#define endl '\n'
const int N = 5011;
vii G(N);
vi visited(N);
int dfs(int now, int par) {
visited[now] = true;
for (int ch: G[now]) {
if (ch == par) continue;
if (visited[ch]) continue;
return dfs(ch, now) + 1;
}
return 1;
}
ll dp[N][N][2];
signed main() {
int n, m;
cin >> n >> m;
vi p(n);
rep (i, n) {
cin >> p[i];
p[i]--;
G[i].push_back(p[i]);
G[p[i]].push_back(i);
}
vi wa;
rep (i, n) {
if (not visited[i]) {
int temp = dfs(i, -1);
wa.push_back(temp);
}
}
dp[0][0][0] = 1;
int i = 0;
for (int cnt: wa) {
for (int now = 0; now < cnt; now++) {
for (int j = 0; j <= i; j++) {
rep (k, 2) {
if (dp[i][j][k] == 0) continue;
if (now == 0) {
// 新しいとこ
(dp[i + 1][j + 1][1] += dp[i][j][k]) %= MOD;
// すでにあるとこ
(dp[i + 1][j][1] += dp[i][j][k] * j % MOD) %= MOD;
}
else if (now != cnt - 1) {
// 新しいとこ
(dp[i + 1][j + 1][0] += dp[i][j][k]) %= MOD;
// すでにある親と違うとこ
if (k == 0) {
(dp[i + 1][j][0] += dp[i][j][k] * (j - 2) % MOD) %= MOD;
}
else {
(dp[i + 1][j][0] += dp[i][j][k] * (j - 1) % MOD) %= MOD;
}
// すでにある親のとこ
if (k == 0) {
(dp[i + 1][j][1] += dp[i][j][k]) %= MOD;
}
}
else {
// 新しいとこ
(dp[i + 1][j + 1][0] += dp[i][j][k]) %= MOD;
// すでにある親と違うとこ
if (k == 0) {
(dp[i + 1][j][0] += dp[i][j][k] * (j - 2) % MOD) %= MOD;
}
else {
(dp[i + 1][j][0] += dp[i][j][k] * (j - 1) % MOD) %= MOD;
}
}
}
}
i++;
}
}
ll ans = dp[n][m][0] + dp[n][m][1];
cout << ans % MOD << endl;
}