結果
問題 | No.2115 Making Forest Easy |
ユーザー | 👑 rin204 |
提出日時 | 2022-10-28 23:43:23 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 3,559 bytes |
コンパイル時間 | 5,542 ms |
コンパイル使用メモリ | 291,464 KB |
実行使用メモリ | 5,244 KB |
最終ジャッジ日時 | 2023-09-20 07:03:08 |
合計ジャッジ時間 | 34,639 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 410 ms
4,528 KB |
testcase_03 | TLE | - |
testcase_04 | AC | 658 ms
4,536 KB |
testcase_05 | AC | 537 ms
4,380 KB |
testcase_06 | AC | 411 ms
4,924 KB |
testcase_07 | AC | 1,175 ms
4,376 KB |
testcase_08 | AC | 415 ms
4,912 KB |
testcase_09 | AC | 580 ms
4,380 KB |
testcase_10 | AC | 415 ms
4,964 KB |
testcase_11 | AC | 1,237 ms
4,380 KB |
testcase_12 | AC | 563 ms
4,384 KB |
testcase_13 | AC | 746 ms
4,376 KB |
testcase_14 | AC | 538 ms
4,380 KB |
testcase_15 | AC | 1,219 ms
4,380 KB |
testcase_16 | AC | 900 ms
4,376 KB |
testcase_17 | AC | 9 ms
4,380 KB |
testcase_18 | AC | 14 ms
4,376 KB |
testcase_19 | AC | 136 ms
4,384 KB |
testcase_20 | AC | 767 ms
4,376 KB |
testcase_21 | AC | 5 ms
4,376 KB |
testcase_22 | AC | 82 ms
4,380 KB |
testcase_23 | AC | 294 ms
4,412 KB |
testcase_24 | AC | 1,095 ms
4,380 KB |
testcase_25 | AC | 130 ms
4,380 KB |
testcase_26 | AC | 545 ms
4,380 KB |
testcase_27 | AC | 422 ms
4,376 KB |
testcase_28 | AC | 1,186 ms
4,380 KB |
testcase_29 | AC | 8 ms
4,380 KB |
testcase_30 | AC | 1,183 ms
4,380 KB |
testcase_31 | AC | 664 ms
4,572 KB |
testcase_32 | AC | 411 ms
4,964 KB |
testcase_33 | AC | 412 ms
4,948 KB |
testcase_34 | AC | 21 ms
4,380 KB |
testcase_35 | AC | 24 ms
4,380 KB |
testcase_36 | AC | 417 ms
5,244 KB |
testcase_37 | AC | 249 ms
4,384 KB |
testcase_38 | AC | 70 ms
4,380 KB |
testcase_39 | AC | 73 ms
4,376 KB |
testcase_40 | AC | 1,152 ms
4,380 KB |
testcase_41 | AC | 547 ms
4,380 KB |
testcase_42 | AC | 1,354 ms
4,380 KB |
testcase_43 | AC | 207 ms
4,380 KB |
testcase_44 | AC | 57 ms
4,380 KB |
testcase_45 | AC | 648 ms
4,380 KB |
testcase_46 | AC | 601 ms
4,380 KB |
testcase_47 | AC | 176 ms
4,384 KB |
testcase_48 | AC | 696 ms
4,584 KB |
testcase_49 | AC | 650 ms
4,820 KB |
testcase_50 | AC | 485 ms
4,380 KB |
testcase_51 | AC | 146 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; 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(), greater<pair<long long, int>>()); 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(auto tmp:A){ a = tmp.first; int i = tmp.second; vector<int> dist(n, -1); vector<int> size(n, 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; size[pos]++; for(auto npos: edges[pos]) size[pos] += size[npos]; } } vector<mint> dp(n, 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[size[npos] - 1]; } else{ self(self, npos, pos); dp[pos] *= dp[npos] + pow2[size[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[size[npos] - 1]); else if(npos == bpos) L[ii + 1] = (L[ii] * dp[npos]); else L[ii + 1] = (L[ii] * (pow2[size[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[size[npos] - 1]); else if(npos == bpos) R[ii] = (R[ii + 1] * dp[npos]); else R[ii] = (R[ii + 1] * (pow2[size[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]; size[pos] = n - size[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(); }