#define _CRT_SECURE_NO_WARNINGS
#include <bits/stdc++.h>
// #include <atcoder/all>
#include <cstdio>
using namespace std;
#ifndef ONLINE_JUDGE
#define dbg(x) cerr << #x <<" "; print(x); cerr << endl;
#else
#define dbg(x)
#endif
#define sz(x) (int((x).size()))
#define len(x) (int)x.length()
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define clr(x) (x).clear()
#define uniq(x) x.resize(unique(all(x)) - x.begin());
#define pb push_back
#define popf pop_front
#define popb pop_back
#define ld long double
#define ll long long
void print(long long t) {cerr << t;}
void print(int t) {cerr << t;}
void print(string t) {cerr << t;}
void print(char t) {cerr << t;}
void print(double t) {cerr << t;}
void print(unsigned long long t) {cerr << t;}
void print(long double t) {cerr << t;}
template <class T, class V> void print(pair <T, V> p);
template <class T> void print(vector <T> v);
template <class T> void print(set <T> v);
template <class T, class V> void print(map <T, V> v);
template <class T> void print(multiset <T> v);
template <class T> void print(T v[],T n) {cerr << "["; for(int i = 0; i < n; i++) {cerr << v[i] << " ";} cerr << "]";}
template <class T, class V> void print(pair <T, V> p) {cerr << "{"; print(p.first); cerr << ","; print(p.second); cerr << "}";}
template <class T> void print(vector <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";}
template <class T> void print(deque <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";}
template <class T> void print(set <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";}
template <class T> void print(multiset <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";}
template <class T, class V> void print(map <T, V> v) {cerr << "[ "; for (auto i : v) {print(i); cerr << " ";} cerr << "]";}
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
#define nl '\n'
// for random generations
mt19937 myrand(chrono::steady_clock::now().time_since_epoch().count());
// mt19937 myrand(131);
// for grid problems
int dx[8] = {-1,0,1,0,1,-1,1,-1};
int dy[8] = {0,1,0,-1,1,1,-1,-1};
long long ka() {
long long x = 0;
bool z = false;
while (1)
{
char y = getchar();
if (y == '-')
z = true;
else if ('0' <= y && y <= '9')
x = x * 10 + y - '0';
else
{
if (z)
x *= -1;
return x;
}
}
}
// lowest / (1 << 17) >= 1e5 / (1 << 18) >= 2e5 / (1 << 21) >= 1e6
void fastIO() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr); cout.tie(nullptr);
}
// file in/out
void setIO(string str = "") {
fastIO();
// if (str != "") {
// freopen((str + ".in").c_str(), "r", stdin);
// freopen((str + ".out").c_str(), "w", stdout);
// }else {
// freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
// }
}
// Indexed Set
template <class T> using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
const int N = 2e5 + 10, M = 7e6 + 10;
vector<int> adj[N];
int n, q, deg[N];
int dp[N][3];
int add[N], depth[N];
int two[N];
bool is_leaf[N];
int sz[N], id[N], tp[N], p[N], dep[N];
struct node {
long long sum, ones, twos, lazy;
};
node combine(node a, node b) {
a.sum += b.sum;
a.ones += b.ones;
a.twos += b.twos;
a.lazy = 0;
return a;
}
struct segTree {
vector<node> tree;
int size = 1;
void propagate(int x, int lx, int rx) {
if(rx - lx == 1 || tree[x].lazy % 2 == 0) {
tree[x].lazy = 0;
return;
}
int mid = (lx + rx) / 2;
tree[2 * x + 1].lazy += tree[x].lazy;
swap(tree[2 * x + 1].ones, tree[2 * x + 1].twos);
tree[2 * x + 1].sum = tree[2 * x + 1].ones + 2 * tree[2 * x + 1].twos;
tree[2 * x + 2].lazy += tree[x].lazy;
swap(tree[2 * x + 2].ones, tree[2 * x + 2].twos);
tree[2 * x + 2].sum = tree[2 * x + 2].ones + 2 * tree[2 * x + 2].twos;
tree[x].lazy = 0;
}
void init(long long s) {
while(size < s) {
size *= 2;
}
tree.assign(2 * size - 1, {0, 0, 0});
}
void upd(int l, int r, long long value, int x, int lx, int rx) {
propagate(x, lx, rx);
if(lx >= r || rx <= l) {
return;
}
if(lx >= l && rx <= r) {
if(value % 2 == 1) {
swap(tree[x].ones, tree[x].twos);
}
tree[x].sum = tree[x].ones + 2 * tree[x].twos;
tree[x].lazy += value;
propagate(x, lx, rx);
return;
}
int mid = (lx + rx) / 2;
upd(l, r, value, 2 * x + 1, lx, mid);
upd(l, r, value, 2 * x + 2, mid, rx);
tree[x] = combine(tree[2 * x + 1], tree[2 * x + 2]);
}
void upd(int l, int r, long long value) {
upd(l, r, value, 0, 0, size);
}
void upd2(int l, int r, long long value, int x, int lx, int rx) {
propagate(x, lx, rx);
if(lx >= r || rx <= l) {
return;
}
if(lx >= l && rx <= r) {
if(value == 1) {
tree[x].ones = 1;
} else {
tree[x].twos = 1;
}
tree[x].sum = tree[x].ones + 2 * tree[x].twos;
tree[x].lazy = 0;
propagate(x, lx, rx);
return;
}
int mid = (lx + rx) / 2;
upd2(l, r, value, 2 * x + 1, lx, mid);
upd2(l, r, value, 2 * x + 2, mid, rx);
tree[x] = combine(tree[2 * x + 1], tree[2 * x + 2]);
}
void upd2(int l, int r, long long value) {
upd2(l, r, value, 0, 0, size);
}
node qry(int l, int r, int x, int lx, int rx) {
propagate(x, lx, rx);
if(lx >= r || rx <= l) {
return {0, 0, 0, 0};
}
if(lx >= l && rx <= r) {
return tree[x];
}
int mid = (lx + rx) / 2;
node s1 = qry(l, r, 2 * x + 1, lx, mid);
node s2 = qry(l, r, 2 * x + 2, mid, rx);
return combine(s1, s2);
}
node qry(int l, int r) {
return qry(l, r, 0, 0, size);
}
};
segTree seg;
int dfs_sz(int cur, int par) {
sz[cur] = 1;
p[cur] = par;
for (int chi : adj[cur]) {
if (chi == par) continue;
dep[chi] = dep[cur] + 1;
p[chi] = cur;
sz[cur] += dfs_sz(chi, cur);
}
return sz[cur];
}
int ct = 1;
void dfs_hld(int cur, int par, int top) {
id[cur] = ct++;
tp[cur] = top;
if(dp[cur][2] != -1) {
seg.upd2(id[cur], id[cur] + 1, 2);
} else {
seg.upd2(id[cur], id[cur] + 1, 1);
}
int h_chi = -1, h_sz = -1;
for (int chi : adj[cur]) {
if (chi == par) continue;
if (sz[chi] > h_sz) {
h_sz = sz[chi];
h_chi = chi;
}
}
if (h_chi == -1) return;
dfs_hld(h_chi, cur, top);
for (int chi : adj[cur]) {
if (chi == par || chi == h_chi) continue;
dfs_hld(chi, cur, chi);
}
}
node path(int x, int y) {
node answ = {0, 0, 0};
int xi = x, yi = y;
while (tp[x] != tp[y]) {
if (dep[tp[x]] < dep[tp[y]]) swap(x, y);
answ = combine(answ, seg.qry(id[tp[x]], id[x] + 1));
x = p[tp[x]];
}
if (dep[x] > dep[y]) swap(x, y);
answ = combine(answ, seg.qry(id[x], id[y] + 1));
return answ;
}
void change_path(int x, int y) {
node answ = {0, 0, 0};
while (tp[x] != tp[y]) {
if (dep[tp[x]] < dep[tp[y]]) swap(x, y);
seg.upd(id[tp[x]], id[x] + 1, 1);
x = p[tp[x]];
}
if (dep[x] > dep[y]) swap(x, y);
seg.upd(id[x], id[y] + 1, 1);
}
void dfs(int node, int parent) {
dp[node][1] = dp[node][2] = -1;
int cnt = 0;
for(auto i: adj[node]) {
if(i == parent) continue;
cnt++;
dfs(i, node);
}
cnt += add[node];
if(!cnt) {
dp[node][1] = 0;
is_leaf[node] = 1;
} else {
long long s = 0, c = 0;
for(auto i: adj[node]) {
if(i == parent) continue;
if(dp[i][1] != -1) {
c += 1;
s += dp[i][1] + 1;
} else if(dp[i][2] != -1) {
c += 2;
s += dp[i][2] + 2;
}
}
s += add[node];
c += add[node];
if(c & 1) {
dp[node][1] = s;
} else {
dp[node][2] = s;
}
}
add[node] = 0;
}
void dfs2(int node, int parent) {
if(parent) {
two[node] += two[parent];
depth[node] = depth[parent] + 1;
if(dp[node][2] != -1) {
two[node]++;
}
}
for(auto i: adj[node]) {
if(i == parent) continue;
dfs2(i, node);
}
}
void solve_() {
// cin >> n >> q;
n = ka(), q = ka();
for(int i = 1; i <= n - 1; i++) {
int a, b;
a = ka(), b = ka();
// cin >> a >> b;
adj[a].push_back(b);
adj[b].push_back(a);
deg[a]++, deg[b]++;
}
int start = -1;
for(int i = 1; i <= n; i++) {
if(deg[i] > 1) {
start = i;
break;
}
}
seg.init(n + 2);
dfs(start, 0);
dfs_sz(start, 0);
dfs_hld(start, 0, 1);
while(q--) {
int d = ka();
long long cur = 0, ans = 0;
vector<int> vec;
vector<pair<int, int>> don;
int ok = 0;
for(int i = 1; i <= d; i++) {
int a = ka();
vec.push_back(a);
deg[a]++;
if(a == start) {
change_path(a, start);
don.push_back({a, start});
cur++;
continue;
} else if(deg[a] == 2) {
continue;
}
cur++;
change_path(a, start);
don.push_back({a, start});
}
if(cur % 2 == 1) {
cout << -1 << '\n';
} else {
cout << seg.qry(2, n + 1).sum + d << '\n';
}
for(auto i: vec) {
deg[i]--;
}
for(auto i: don) {
change_path(i.first, i.second);
}
}
}
int main() {
setIO();
auto solve = [&](int test_case)-> void {
while(test_case--) {
solve_();
}
};
int test_cases = 1;
// cin >> test_cases;
solve(test_cases);
return 0;
}
Compilation message
cleaning.cpp: In member function 'void segTree::propagate(int, int, int)':
cleaning.cpp:131:9: warning: unused variable 'mid' [-Wunused-variable]
131 | int mid = (lx + rx) / 2;
| ^~~
cleaning.cpp: In function 'node path(int, int)':
cleaning.cpp:272:7: warning: unused variable 'xi' [-Wunused-variable]
272 | int xi = x, yi = y;
| ^~
cleaning.cpp:272:15: warning: unused variable 'yi' [-Wunused-variable]
272 | int xi = x, yi = y;
| ^~
cleaning.cpp: In function 'void change_path(int, int)':
cleaning.cpp:284:8: warning: unused variable 'answ' [-Wunused-variable]
284 | node answ = {0, 0, 0};
| ^~~~
cleaning.cpp: In function 'void solve_()':
cleaning.cpp:382:24: warning: unused variable 'ans' [-Wunused-variable]
382 | long long cur = 0, ans = 0;
| ^~~
cleaning.cpp:385:9: warning: unused variable 'ok' [-Wunused-variable]
385 | int ok = 0;
| ^~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
5076 KB |
Output is correct |
| 2 |
Incorrect |
179 ms |
8668 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
85 ms |
7032 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
116 ms |
7740 KB |
Output is correct |
| 2 |
Correct |
94 ms |
7748 KB |
Output is correct |
| 3 |
Correct |
91 ms |
27932 KB |
Output is correct |
| 4 |
Correct |
288 ms |
28028 KB |
Output is correct |
| 5 |
Correct |
107 ms |
26652 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
153 ms |
9272 KB |
Output is correct |
| 2 |
Incorrect |
69 ms |
8592 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
290 ms |
18236 KB |
Output is correct |
| 2 |
Correct |
432 ms |
18064 KB |
Output is correct |
| 3 |
Correct |
321 ms |
11592 KB |
Output is correct |
| 4 |
Correct |
393 ms |
18068 KB |
Output is correct |
| 5 |
Correct |
508 ms |
18068 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
456 ms |
21024 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
5076 KB |
Output is correct |
| 2 |
Incorrect |
179 ms |
8668 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
