Submission #1956524
Source Code Expand
#define _CRT_SECURE_NO_WARNINGS #include <algorithm> #include <array> #include <cmath> #include <cstdio> #include <vector> using namespace std; const int MAX_V = 100000 * 2; const int MAX_LOG_V = (int)log2(MAX_V); const int MAX_Q = 100000; //vector<int> G[MAX_V] array<vector<int>, MAX_V> G; // グラフの隣接リスト表現 int root; // 根ノードの番号 // 親を 2^k 回辿って到達する頂点(根を通りすぎる場合は -1 とする) // int parent[MAX_LOG_V][MAX_V] vector< vector<int> > parent(MAX_LOG_V, vector<int>(MAX_V, -1)); // 根からの深さ // int depth[MAX_V] array<int, MAX_V> depth; // q[MAX_Q][2] array<array<int, 2>, MAX_Q> q; template<class T> void gc(T &x) { char c; int s = 0; x = 0; c = getchar_unlocked(); if (c == '-') s = 1; else if ('0' <= c && c <= '9') x = c - '0'; while (true) { c = getchar_unlocked(); if (c < '0' || c > '9') break; x = x * 10 + (c - '0'); } if (s) x = -x; } void dfs(int v, int p, int d) { parent[0][v] = p; depth[v] = d; for (int i = 0; i < (signed)G[v].size(); i++) if (G[v][i] != p) dfs(G[v][i], v, d + 1); } // 初期化 void init(int V) { // parent[0] と depth を初期化する dfs(root, -1, 0); // parent を初期化する for (int k = 0; k + 1 < MAX_LOG_V; k++) for (int v = 0; v < V; v++) if (parent[k][v] < 0) parent[k + 1][v] = -1; else parent[k + 1][v] = parent[k][parent[k][v]]; } // u と v の LCA を求める int lca(int u, int v) { // u と v の深さが同じになるまで親を辿る if (depth[u] > depth[v]) swap(u, v); for (int k = 0; k < MAX_LOG_V; k++) if ((depth[v] - depth[u]) >> k & 1) v = parent[k][v]; if (u == v) return u; // 二分探索で LCA を求める for (int k = MAX_LOG_V - 1; 0 <= k; k--) if (parent[k][u] != parent[k][v]) { u = parent[k][u]; v = parent[k][v]; } return parent[0][u]; } int main() { int N; gc(N); for (int i = 0; i < N - 1; i++) { int x, y; gc(x); gc(y); x--; y--; G[x].push_back(y); G[y].push_back(x); } int Q; gc(Q); root = 0; init(N); for (int i = 0; i < Q; i++) { int a, b; gc(a); gc(b); a--; b--; int l = lca(a, b); printf("%d\n", depth[a] + depth[b] - 2 * depth[l] + 1); } return 0; }
Submission Info
Submission Time | |
---|---|
Task | D - 閉路 |
User | ShinjiSHIBATA |
Language | C++14 (GCC 5.4.1) |
Score | 100 |
Code Size | 2697 Byte |
Status | AC |
Exec Time | 107 ms |
Memory | 28656 KB |
Judge Result
Set Name | Sample | Subtask1 | Subtask2 | ||||||
---|---|---|---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 30 / 30 | 70 / 70 | ||||||
Status |
|
|
|
Set Name | Test Cases |
---|---|
Sample | subtask0_sample01.txt, subtask0_sample02.txt, subtask0_sample03.txt |
Subtask1 | subtask1_01.txt, subtask1_02.txt, subtask1_03.txt, subtask1_04.txt, subtask1_05.txt, subtask1_06.txt, subtask1_07.txt, subtask1_08.txt, subtask1_09.txt, subtask1_10.txt, subtask1_11.txt, subtask1_12.txt |
Subtask2 | subtask0_sample01.txt, subtask0_sample02.txt, subtask0_sample03.txt, subtask1_01.txt, subtask1_02.txt, subtask1_03.txt, subtask1_04.txt, subtask1_05.txt, subtask1_06.txt, subtask1_07.txt, subtask1_08.txt, subtask1_09.txt, subtask1_10.txt, subtask1_11.txt, subtask1_12.txt, subtask2_01.txt, subtask2_02.txt, subtask2_03.txt, subtask2_04.txt, subtask2_05.txt, subtask2_06.txt, subtask2_07.txt, subtask2_08.txt, subtask2_09.txt, subtask2_10.txt, subtask2_11.txt, subtask2_12.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
subtask0_sample01.txt | AC | 9 ms | 19072 KB |
subtask0_sample02.txt | AC | 9 ms | 19072 KB |
subtask0_sample03.txt | AC | 9 ms | 19072 KB |
subtask1_01.txt | AC | 26 ms | 28016 KB |
subtask1_02.txt | AC | 27 ms | 28016 KB |
subtask1_03.txt | AC | 9 ms | 19072 KB |
subtask1_04.txt | AC | 9 ms | 19072 KB |
subtask1_05.txt | AC | 9 ms | 19072 KB |
subtask1_06.txt | AC | 9 ms | 19072 KB |
subtask1_07.txt | AC | 33 ms | 21872 KB |
subtask1_08.txt | AC | 34 ms | 21872 KB |
subtask1_09.txt | AC | 34 ms | 21872 KB |
subtask1_10.txt | AC | 34 ms | 21872 KB |
subtask1_11.txt | AC | 34 ms | 21872 KB |
subtask1_12.txt | AC | 34 ms | 21872 KB |
subtask2_01.txt | AC | 40 ms | 28656 KB |
subtask2_02.txt | AC | 40 ms | 28656 KB |
subtask2_03.txt | AC | 25 ms | 19072 KB |
subtask2_04.txt | AC | 28 ms | 19072 KB |
subtask2_05.txt | AC | 34 ms | 19072 KB |
subtask2_06.txt | AC | 33 ms | 19072 KB |
subtask2_07.txt | AC | 95 ms | 22128 KB |
subtask2_08.txt | AC | 100 ms | 22128 KB |
subtask2_09.txt | AC | 105 ms | 22128 KB |
subtask2_10.txt | AC | 106 ms | 22256 KB |
subtask2_11.txt | AC | 107 ms | 22256 KB |
subtask2_12.txt | AC | 107 ms | 22256 KB |