oj.uz

Submission #977780

# Submission time Handle Problem Language Result Execution time Memory
977780 2024-05-08T11:02:37 Z OAleksa Spring cleaning (CEOI20_cleaning) C++14
44 / 100
 1000 ms 22612 KB 
#include <bits/stdc++.h>
#define f first
#define s second
using namespace std;
const int N = 1e5 + 69;
const int K = 17;
pair<int, int> st[4 * N];
int n, q, ch[N], par[N], leaf[N], is[N], dep[N];
int tin[N], timer, top[N], sz[N], lzy[N];
vector<int> g[N];
void push(int v) {
	if (lzy[v] % 2 == 1) {
		lzy[v * 2] += lzy[v];
		lzy[v * 2 + 1] += lzy[v];
		swap(st[v * 2].f, st[v * 2].s);
		swap(st[v * 2 + 1].f, st[v * 2 + 1].s);
		lzy[v] = 0;
	}
}
void dfs(int v, int p) {
	par[v] = p;
	sz[v] = 1;
	if (g[v].size() == 1)
		leaf[v] = ch[v] = is[v] = 1;
	for (auto u : g[v]) {
		if (u == p)
			continue;
		dep[u] = dep[v] + 1;
		dfs(u, v);
		ch[v] += ch[u];
		sz[v] += sz[u];
	}
}
void modify(int v, int tl, int tr, int p, int x) {
	if (tl == tr) {
		if (x % 2 == 0)
			st[v] = {0, 1};
		else
			st[v] = {1, 0};
	}
	else {
		int mid = (tl + tr) / 2;
		if (p <= mid)
			modify(v * 2, tl, mid, p, x);
		else
			modify(v * 2 + 1, mid + 1, tr, p, x);
		st[v] = {st[v * 2].f + st[v * 2 + 1].f, st[v * 2].s + st[v * 2 + 1].s};
	}
}
void svapuj(int v, int tl, int tr, int l, int r) {
	if (tl > r || tr < l)
		return;
	else if (tl >= l && tr <= r) {
		lzy[v]++;
		swap(st[v].f, st[v].s);
	}
	else {
		push(v);
		int mid = (tl + tr) / 2;
		svapuj(v * 2, tl, mid, l, r);
		svapuj(v * 2 + 1, mid + 1, tr, l, r);
		st[v] = {st[v * 2].f + st[v * 2 + 1].f, st[v * 2].s + st[v * 2 + 1].s};
	}
}
pair<int, int> get(int v, int tl, int tr, int l, int r) {
	if (tl > r || tr < l)
		return {0, 0};
	else if (tl >= l && tr <= r)
		return st[v];
	else {
		int mid = (tl + tr) / 2;
		push(v);
		auto ll = get(v * 2, tl, mid, l, r);
		auto rr = get(v * 2 + 1, mid + 1, tr, l, r);
		pair<int, int> res;
		res.f = {ll.f + rr.f};
		res.s = {ll.s + rr.s};
		return res;
	}
}
void hld(int v, int p, int tp) {
	int s = -1;
	tin[v] = ++timer;
	top[v] = tp;
	modify(1, 1, n, tin[v], ch[v]);
	for (auto u : g[v]) {
		if (u == p)
			continue;
		if (s == -1 || sz[s] < sz[u])
			s = u;
	}
	if (s != -1)
		hld(s, v, tp);
	for (auto u : g[v]) {
		if (u == p || u == s)
			continue;
		hld(u, v, u);
	}
}
signed main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> n >> q;
	for (int i = 1;i <= n - 1;i++) {
		int a, b;
		cin >> a >> b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	dfs(1, 0);
	hld(1, 0, 1);
	int uk = ch[1];
	while (q--) {
		int t;
		cin >> t;
		int a[t];
		map<int, int> cnt;
		for (int i = 0;i < t;i++) {
			cin >> a[i];
			cnt[a[i]]++;
			uk++;
			if (leaf[a[i]]) {
				leaf[a[i]]--;
				uk--;
			}
		}
		for (auto u : cnt) {
			int ok = 0, v = u.f;
			if (is[v] && u.s % 2 == 0) 
				ok = 1;
			else if (!is[v] && u.s % 2 == 1) 
				ok = 1;
			if (ok) {
				while (v > 0) {
					svapuj(1, 1, n, tin[top[v]], tin[v]);
					v = par[top[v]];
				}
			}
		}
		if (uk % 2 == 1)
			cout << "-1\n";
		else {
			pair<int, int> res;
			res = get(1, 1, n, 2, n);
			cout << res.f + 2 * res.s + t << '\n';
		}
		for (int i = 0;i < t;i++) {
			uk--;
			if (is[a[i]] && !leaf[a[i]]) {
				uk++;
				leaf[a[i]] = 1;
			}
		}
		for (auto u : cnt) {
			int ok = 0, v = u.f;
			if (is[v] && u.s % 2 == 0) 
				ok = 1;
			else if (!is[v] && u.s % 2 == 1) 
				ok = 1;
			if (ok) {
				while (v > 0) {
					svapuj(1, 1, n, tin[top[v]], tin[v]);
					v = par[top[v]];
				}
			}
		}
	}
	return 0;
}

Subtask #1
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 6744 KB Output is correct
2 Correct 77 ms 9564 KB Output is correct

Subtask #2
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 7256 KB Output is correct
2 Correct 14 ms 7260 KB Output is correct
3 Correct 30 ms 12492 KB Output is correct
4 Incorrect 62 ms 14788 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 7516 KB Output is correct
2 Correct 14 ms 7512 KB Output is correct
3 Correct 45 ms 19536 KB Output is correct
4 Correct 77 ms 22612 KB Output is correct
5 Correct 38 ms 19548 KB Output is correct

Subtask #4
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 40 ms 10188 KB Output is correct
2 Correct 30 ms 9564 KB Output is correct
3 Correct 9 ms 9308 KB Output is correct
4 Correct 9 ms 9888 KB Output is correct
5 Correct 10 ms 10196 KB Output is correct
6 Correct 42 ms 9916 KB Output is correct

Subtask #5
19.0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 94 ms 11092 KB Output is correct
2 Correct 155 ms 10872 KB Output is correct
3 Correct 121 ms 10008 KB Output is correct
4 Correct 153 ms 10888 KB Output is correct
5 Correct 153 ms 10832 KB Output is correct

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1065 ms 12884 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #7
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 6744 KB Output is correct
2 Correct 77 ms 9564 KB Output is correct
3 Correct 12 ms 7256 KB Output is correct
4 Correct 14 ms 7260 KB Output is correct
5 Correct 30 ms 12492 KB Output is correct
6 Incorrect 62 ms 14788 KB Output isn't correct
7 Halted 0 ms 0 KB -