結果
問題 | No.1718 Random Squirrel |
ユーザー | 00 Sakuda |
提出日時 | 2024-03-31 22:59:33 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 408 ms / 2,000 ms |
コード長 | 2,322 bytes |
コンパイル時間 | 2,357 ms |
コンパイル使用メモリ | 212,588 KB |
実行使用メモリ | 156,424 KB |
最終ジャッジ日時 | 2024-03-31 22:59:45 |
合計ジャッジ時間 | 11,894 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 48 ms
128,392 KB |
testcase_01 | AC | 48 ms
128,392 KB |
testcase_02 | AC | 48 ms
128,392 KB |
testcase_03 | AC | 48 ms
128,392 KB |
testcase_04 | AC | 50 ms
128,392 KB |
testcase_05 | AC | 49 ms
128,392 KB |
testcase_06 | AC | 48 ms
128,392 KB |
testcase_07 | AC | 48 ms
128,392 KB |
testcase_08 | AC | 49 ms
128,392 KB |
testcase_09 | AC | 48 ms
128,392 KB |
testcase_10 | AC | 49 ms
128,392 KB |
testcase_11 | AC | 229 ms
130,568 KB |
testcase_12 | AC | 82 ms
128,776 KB |
testcase_13 | AC | 167 ms
129,672 KB |
testcase_14 | AC | 237 ms
130,696 KB |
testcase_15 | AC | 243 ms
130,312 KB |
testcase_16 | AC | 129 ms
129,288 KB |
testcase_17 | AC | 217 ms
130,568 KB |
testcase_18 | AC | 315 ms
131,080 KB |
testcase_19 | AC | 221 ms
130,184 KB |
testcase_20 | AC | 110 ms
129,160 KB |
testcase_21 | AC | 316 ms
131,592 KB |
testcase_22 | AC | 393 ms
131,592 KB |
testcase_23 | AC | 323 ms
131,592 KB |
testcase_24 | AC | 323 ms
131,592 KB |
testcase_25 | AC | 408 ms
131,592 KB |
testcase_26 | AC | 236 ms
131,880 KB |
testcase_27 | AC | 237 ms
131,848 KB |
testcase_28 | AC | 234 ms
131,880 KB |
testcase_29 | AC | 279 ms
151,816 KB |
testcase_30 | AC | 291 ms
156,424 KB |
testcase_31 | AC | 277 ms
151,816 KB |
testcase_32 | AC | 278 ms
151,816 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; ll INF = (1LL << 60); int MN = 2e6; vector<vector<int>> G(MN); vector<ll> dp1(MN, 0), DP1(MN, 0); vector<ll> dp2(MN, 0), DP2(MN, 0); vector<int> A(MN, 0); vector<int> PAR(MN, -1); void dfs1(int v, int par) { PAR[v] = par; for (auto nv : G[v]) { if (nv == par) continue; dfs1(nv, v); if (A[nv] + dp1[nv] > 0) dp1[v] += 2 + dp1[nv]; } } void dfs2(int v, int par) { ll s = 0; for (auto nv : G[v]) { if (nv == par) { if (A[nv] + DP1[v] > 0) s += 2 + DP1[v]; } else { if (A[nv] + dp1[nv] > 0) s += 2 + dp1[nv]; } } for (auto nv : G[v]) { if (nv == par) continue; DP1[nv] = s; if (A[nv] + dp1[nv] > 0) DP1[nv] -= 2 + dp1[nv]; dfs2(nv, v); } } void DFS(int v, int par) { ll mins = INF; dp2[v] = dp1[v]; for (auto nv : G[v]) { if (nv == par) continue; DFS(nv, v); if (A[nv] + dp1[nv] > 0) { mins = min(mins, dp2[nv] - dp1[nv] - 1); } } if (mins != INF) dp2[v] += mins; } void DFS2(int v, int par) { multiset<ll> sets; ll s = 0; for (auto nv : G[v]) { if (nv == par) { if (A[nv] + DP1[v] == 0) continue; s += 2 + DP1[v]; sets.insert(DP2[v] - DP1[v] - 1); } else { if (A[nv] + dp1[nv] == 0) continue; s += 2 + dp1[nv]; sets.insert(dp2[nv] - dp1[nv] - 1); } } for (auto nv : G[v]) { if (nv == par) continue; DP2[nv] = s; if (A[nv] + dp1[nv] > 0) { DP2[nv] -= 2 + dp1[nv]; sets.erase(sets.find(dp2[nv] - dp1[nv] - 1)); } if (!sets.empty()) { DP2[nv] += *sets.begin(); } if (A[nv] + dp1[nv] > 0) sets.insert(dp2[nv] - dp1[nv] - 1); DFS2(nv, v); } } int main() { int N, K;cin >> N >> K; for (int i = 0;i < N - 1;i++) { int u, v;cin >> u >> v; u--;v--; G[u].push_back(v); G[v].push_back(u); } for (int i = 0;i < K;i++) { int d;cin >> d; d--; A[d] = 1; } int r; r = 0; //cin >> r; dfs1(r, r); dfs2(r, r); DFS(r, r); DFS2(r, r); for (int v = 0;v < N;v++) { ll ans = 0; ll mins = INF; for (auto nv : G[v]) { if (nv == PAR[v]) { if (A[nv] + DP1[v] > 0) { ans += 2 + DP1[v]; mins = min(mins, DP2[v] - DP1[v] - 1); } } else { if (A[nv] + dp1[nv] > 0) { ans += 2 + dp1[nv]; mins = min(mins, dp2[nv] - dp1[nv] - 1); } } } if (mins != INF) ans += mins; cout << ans << endl; } }