結果
問題 | No.1769 Don't Stop the Game |
ユーザー | 👑 Nachia |
提出日時 | 2021-11-27 00:34:01 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 213 ms / 3,000 ms |
コード長 | 2,755 bytes |
コンパイル時間 | 2,392 ms |
コンパイル使用メモリ | 93,640 KB |
実行使用メモリ | 32,904 KB |
最終ジャッジ日時 | 2023-09-12 06:11:37 |
合計ジャッジ時間 | 7,220 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,380 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 2 ms
4,384 KB |
testcase_03 | AC | 2 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,380 KB |
testcase_05 | AC | 2 ms
4,376 KB |
testcase_06 | AC | 2 ms
4,380 KB |
testcase_07 | AC | 2 ms
4,380 KB |
testcase_08 | AC | 149 ms
19,980 KB |
testcase_09 | AC | 92 ms
14,416 KB |
testcase_10 | AC | 142 ms
19,284 KB |
testcase_11 | AC | 101 ms
14,268 KB |
testcase_12 | AC | 193 ms
24,220 KB |
testcase_13 | AC | 201 ms
23,996 KB |
testcase_14 | AC | 211 ms
24,536 KB |
testcase_15 | AC | 196 ms
23,100 KB |
testcase_16 | AC | 192 ms
23,272 KB |
testcase_17 | AC | 199 ms
22,880 KB |
testcase_18 | AC | 203 ms
23,004 KB |
testcase_19 | AC | 211 ms
22,656 KB |
testcase_20 | AC | 212 ms
23,868 KB |
testcase_21 | AC | 212 ms
24,308 KB |
testcase_22 | AC | 213 ms
22,940 KB |
testcase_23 | AC | 174 ms
23,364 KB |
testcase_24 | AC | 190 ms
22,752 KB |
testcase_25 | AC | 93 ms
26,324 KB |
testcase_26 | AC | 106 ms
26,072 KB |
testcase_27 | AC | 125 ms
32,904 KB |
testcase_28 | AC | 135 ms
32,904 KB |
testcase_29 | AC | 130 ms
28,312 KB |
testcase_30 | AC | 135 ms
28,132 KB |
ソースコード
#include <iostream> #include <vector> #include <algorithm> using namespace std; using i32 = int32_t; using u32 = uint32_t; using i64 = int64_t; using u64 = uint64_t; #define rep(i,n) for(int i=0; i<(n); i++) class CoordinateCompress { using Elem = i64; static const Elem negInf = -1001001001001001001; vector<pair<Elem, int>> G; vector<int> res; bool ok = true; public: int push(Elem x) { ok = false; G.push_back({ x,(int)G.size() }); return G.back().second; } void calc() { if (ok) return; sort(G.begin(), G.end()); res.resize(G.size()); Elem x = negInf; int p = -1; rep(i, G.size()) { if (x != G[i].first) { x = G[i].first; p++; } res[G[i].second] = p; } ok = true; } int operator[](int i) { calc(); return res[i]; } }; struct Edge{ int to; u32 h; }; int N; vector<vector<Edge>> E; vector<int> P; vector<int> Z; vector<u32> H; // vector<vector<int>> black_stack; vector<int> black_stack; vector<int> prev_black; vector<int> block_size; vector<int> root_size; i64 ans; void dfs(int p = 0){ block_size[p] = Z[p]; prev_black[p] = black_stack[H[p]]; if(prev_black[p] != 0) block_size[prev_black[p]] -= Z[p]; else root_size[H[p]] -= Z[p]; for(auto e : E[p]) if(e.to != P[p]){ int black_stack_tmp = black_stack[H[p]]; black_stack[H[p]] = e.to; dfs(e.to); black_stack[H[p]] = black_stack_tmp; } } int main(){ cin >> N; E.resize(N); rep(i,N-1){ int u,v; u32 c; cin >> u >> v >> c; u--; v--; E[u].push_back({ v,c }); E[v].push_back({ u,c }); } vector<int> I = {0}; P.assign(N,-1); H.assign(N,0); rep(i,I.size()){ int p = I[i]; for(auto e : E[p]){ if(P[p] == e.to) continue; I.push_back(e.to); P[e.to] = p; H[e.to] = H[p] ^ e.h; } } Z.assign(N,1); for(int i=N-1; i>=1; i--) Z[P[I[i]]] += Z[I[i]]; { CoordinateCompress CCH; rep(i,N) H[i] = CCH.push(H[i]); CCH.calc(); rep(i,N) H[i] = CCH[H[i]]; } black_stack.assign(N,0); prev_black.assign(N,0); block_size.assign(N,0); root_size.assign(N,N); ans = 0; dfs(); i64 ans = 0; for(int p=1; p<N; p++) ans += block_size[p]; for(int p=1; p<N; p++){ if(prev_black[p] != 0) ans += block_size[prev_black[p]]; else ans += root_size[H[p]]; } cout << ans << endl; return 0; } struct ios_do_not_sync { ios_do_not_sync() { ios::sync_with_stdio(false); cin.tie(nullptr); } } ios_do_not_sync_instance;