結果
問題 | No.2115 Making Forest Easy |
ユーザー | 👑 rin204 |
提出日時 | 2022-10-28 23:48:45 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 3,642 bytes |
コンパイル時間 | 5,432 ms |
コンパイル使用メモリ | 290,684 KB |
実行使用メモリ | 5,116 KB |
最終ジャッジ日時 | 2023-09-20 07:10:32 |
合計ジャッジ時間 | 34,636 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 425 ms
4,604 KB |
testcase_03 | TLE | - |
testcase_04 | AC | 636 ms
4,428 KB |
testcase_05 | AC | 515 ms
4,380 KB |
testcase_06 | AC | 422 ms
4,712 KB |
testcase_07 | AC | 1,141 ms
4,376 KB |
testcase_08 | AC | 423 ms
5,024 KB |
testcase_09 | AC | 588 ms
4,376 KB |
testcase_10 | AC | 424 ms
5,040 KB |
testcase_11 | AC | 1,286 ms
4,376 KB |
testcase_12 | AC | 576 ms
4,380 KB |
testcase_13 | AC | 786 ms
4,380 KB |
testcase_14 | AC | 552 ms
4,380 KB |
testcase_15 | AC | 1,192 ms
4,380 KB |
testcase_16 | AC | 947 ms
4,380 KB |
testcase_17 | AC | 9 ms
4,380 KB |
testcase_18 | AC | 14 ms
4,376 KB |
testcase_19 | AC | 136 ms
4,380 KB |
testcase_20 | AC | 753 ms
4,376 KB |
testcase_21 | AC | 5 ms
4,376 KB |
testcase_22 | AC | 83 ms
4,376 KB |
testcase_23 | AC | 291 ms
4,376 KB |
testcase_24 | AC | 1,138 ms
4,376 KB |
testcase_25 | AC | 124 ms
4,376 KB |
testcase_26 | AC | 554 ms
4,380 KB |
testcase_27 | AC | 428 ms
4,380 KB |
testcase_28 | AC | 1,146 ms
4,376 KB |
testcase_29 | AC | 8 ms
4,380 KB |
testcase_30 | AC | 1,155 ms
4,380 KB |
testcase_31 | AC | 645 ms
4,608 KB |
testcase_32 | AC | 425 ms
5,032 KB |
testcase_33 | AC | 424 ms
5,076 KB |
testcase_34 | AC | 22 ms
4,380 KB |
testcase_35 | AC | 25 ms
4,380 KB |
testcase_36 | AC | 424 ms
5,116 KB |
testcase_37 | AC | 257 ms
4,376 KB |
testcase_38 | AC | 73 ms
4,376 KB |
testcase_39 | AC | 73 ms
4,380 KB |
testcase_40 | AC | 1,121 ms
4,376 KB |
testcase_41 | AC | 560 ms
4,376 KB |
testcase_42 | AC | 1,396 ms
4,380 KB |
testcase_43 | AC | 194 ms
4,380 KB |
testcase_44 | AC | 59 ms
4,380 KB |
testcase_45 | AC | 618 ms
4,376 KB |
testcase_46 | AC | 646 ms
4,376 KB |
testcase_47 | AC | 161 ms
4,376 KB |
testcase_48 | AC | 665 ms
4,512 KB |
testcase_49 | AC | 618 ms
4,916 KB |
testcase_50 | AC | 494 ms
4,376 KB |
testcase_51 | AC | 151 ms
4,380 KB |
ソースコード
#pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include<bits/stdc++.h> #include<atcoder/modint> using namespace std; using namespace atcoder; using mint = modint998244353; const int N = 5050; int dist[N]; int siz[N]; mint dp[N]; void solve(){ int n; cin >> n; vector<pair<long long, int>> A(n); long long a; for(int i = 0; i < n; i++){ cin >> a; A[i] = {a, i}; } vector<vector<int>> edges(n, vector<int>()); int u, v; for(int i = 0; i < n - 1; i++){ cin >> u >> v; u--; v--; edges[u].push_back(v); edges[v].push_back(u); } sort(A.begin(), A.end()); vector<bool> used(n, false); vector<mint> pow2(n + 1, 1); for(int i = 1; i <= n; i++) pow2[i] = pow2[i - 1] * 2; mint ans = 0; for(int jj = n - 1; jj >= 0; jj--){ a = A[jj].first; int i = A[jj].second; for(int i = 0; i < n; i++){ dist[i] = -1; siz[i] = 0; } dist[i] = 0; stack<int> st; st.push(~i); st.push(i); while(!st.empty()){ int pos = st.top(); st.pop(); if(pos >= 0){ for(auto npos: edges[pos]){ if(dist[npos] == -1){ dist[npos] = dist[pos] + 1; st.push(~npos); st.push(npos); } } } else{ pos = ~pos; siz[pos]++; for(auto npos: edges[pos]) siz[pos] += siz[npos]; } } for(int i = 0; i < n; i++) dp[i] = 0; auto dfs=[&](auto self, int pos, int bpos) -> void { dp[pos] = 1; for(auto npos: edges[pos]){ if(npos == bpos) continue; if(used[npos]){ dp[pos] *= pow2[siz[npos] - 1]; } else{ self(self, npos, pos); dp[pos] *= dp[npos] + pow2[siz[npos] - 1]; } } }; dfs(dfs, i, -1); mint tot = 0; auto dfs2=[&](auto self, int pos, int bpos) -> void { int le = edges[pos].size(); vector<mint> L(le + 1, 1); for(int ii = 0; ii < le; ii++){ int npos = edges[pos][ii]; if(used[npos]) L[ii + 1] = (L[ii] * pow2[siz[npos] - 1]); else if(npos == bpos) L[ii + 1] = (L[ii] * dp[npos]); else L[ii + 1] = (L[ii] * (pow2[siz[npos] - 1] + dp[npos])); } tot += L[le]; vector<mint> R(le + 1, 1); for(int ii = edges[pos].size() - 1; ii >= 0; ii--){ int npos = edges[pos][ii]; if(used[npos]) R[ii] = (R[ii + 1] * pow2[siz[npos] - 1]); else if(npos == bpos) R[ii] = (R[ii + 1] * dp[npos]); else R[ii] = (R[ii + 1] * (pow2[siz[npos] - 1] + dp[npos])); } for(int ii = 0; ii < edges[pos].size(); ii++){ int npos = edges[pos][ii]; if((npos == bpos) || used[npos]) continue; dp[pos] = L[ii] * R[ii + 1]; siz[pos] = n - siz[npos]; self(self, npos, pos); } }; dfs2(dfs2, i, -1); ans += tot * a; used[i] = true; } cout << ans.val() << "\n"; } int main(){ cin.tie(0)->sync_with_stdio(0); int t; t = 1; // cin >> t; while(t--) solve(); }