結果
問題 | 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; }