-
Notifications
You must be signed in to change notification settings - Fork 18
/
Copy path1136.cpp
130 lines (114 loc) · 3.71 KB
/
1136.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#include <bits/stdc++.h>
using namespace std;
#define print_op(...) ostream& operator<<(ostream& out, const __VA_ARGS__& u)
template<typename A, typename B> print_op(pair<A, B>) { return out << "(" << u.first << ", " << u.second << ")"; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> print_op(T_container) { out << "{"; string sep; for (const T &x : u) out << sep << x, sep = ", "; return out << "}"; }
template<typename T> void dbg_out(string s, T x) {cerr << "\033[1;35m" << s << "\033[0;32m = \033[33m" << x << "\033[0m\n";}
template<typename T, typename... Args> void dbg_out(string s, T x, Args... args) {for (int i=0, b=0; i<(int)s.size(); i++) if (s[i] == '(' || s[i] == '{') b++; else
if (s[i] == ')' || s[i] == '}') b--; else if (s[i] == ',' && b == 0) {cerr << "\033[1;35m" << s.substr(0, i) << "\033[0;32m = \033[33m" << x << "\033[31m | "; dbg_out(s.substr(s.find_first_not_of(' ', i + 1)), args...); break;}}
#ifdef LOCAL
#define dbg(...) dbg_out(#__VA_ARGS__, __VA_ARGS__)
#else
#define dbg(...)
#endif
#define ar array
#define ll long long
#define ld long double
#define sz(x) ((int)x.size())
#define rep(i, a, b) for (int i = (int)(a); i < (int)(b); i++)
#define all(a) (a).begin(), (a).end()
const int MAX_N = 1e5 + 5;
const int MAX_L = 20;
const int MAX_C = 26;
const ll MOD = 1e9 + 7;
const ll INF = 1e9;
const ld EPS = 1e-9;
struct BinaryLift {
int n, lg;
vector<int> dep, cnt;
vector<vector<int>> adj, par;
BinaryLift(int _n) : n(_n), lg(__lg(n - 1) + 1), dep(n), cnt(n), adj(n), par(n, vector<int>(lg, -1)) {}
void addEdge(int u, int v) {
adj[u].push_back(v);
adj[v].push_back(u);
}
void init(int s = 0) {
dfs(s);
for (int j = 1; j < lg; j++) {
for (int i = 0; i < n; i++) {
if (par[i][j - 1] != -1) {
par[i][j] = par[par[i][j - 1]][j - 1];
}
}
}
}
void dfs(int u) {
for (int v : adj[u]) {
if (v == par[u][0]) continue;
dep[v] = dep[u] + 1;
par[v][0] = u;
dfs(v);
}
}
int kthAnc(int u, int k) {
if (k > dep[u]) return -2;
for (int j = 0; j < lg; j++) {
if (k & (1 << j)) {
u = par[u][j];
}
}
return u;
}
int lca(int u, int v) {
if (dep[u] < dep[v]) swap(u, v);
u = kthAnc(u, dep[u] - dep[v]);
if (u == v) return u;
for (int j = lg - 1; j >= 0; j--) {
if (par[u][j] != par[v][j]) {
u = par[u][j];
v = par[v][j];
}
}
return par[u][0];
}
int dist(int u, int v) {
return dep[u] + dep[v] - 2 * dep[lca(u, v)];
}
void addPath(int u, int v) {
int m = lca(u, v);
cnt[u]++; cnt[v]++; cnt[m]--;
if (par[m][0] != -1) cnt[par[m][0]]--;
}
void dfsPath(int u, int p = -1) {
for (int v : adj[u]) {
if (v == p) continue;
dfsPath(v, u);
cnt[u] += cnt[v];
}
}
};
void solve(int tc = 0) {
int n, q; cin >> n >> q;
BinaryLift bl(n);
for (int i = 0; i < n - 1; i++) {
int u, v; cin >> u >> v; u--; v--;
bl.addEdge(u, v);
}
bl.init(0);
while (q--) {
int u, v; cin >> u >> v; u--; v--;
bl.addPath(u, v);
}
bl.dfsPath(0);
for (int x : bl.cnt) cout << x << " ";
cout << "\n";
}
signed main() {
ios_base::sync_with_stdio(false); cin.tie(NULL);
int tc = 1;
// cin >> tc;
for (int t = 1; t <= tc; t++) {
// cout << "Case #" << t << ": ";
solve(t);
}
}