oj.uz

답안 #390510

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
390510 2021-04-16T08:38:40 Z talant117408 Spring cleaning (CEOI20_cleaning) C++17
34 / 100
 368 ms 15996 KB 
/*
    Code written by Talant I.D.
*/
#include <bits/stdc++.h>
 
using namespace std;

typedef long long ll;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
 
#define precision(n) fixed << setprecision(n)
#define pb push_back
#define ub upper_bound
#define lb lower_bound
#define mp make_pair
#define eps (double)1e-9
#define PI 2*acos(0.0)
#define endl "\n"
#define sz(v) int((v).size())
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define do_not_disturb ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define OK cout << "OK" << endl;
 
const int mod = 1e9+7;
 
ll mode(ll a) {
    a %= mod;
    if (a < 0) a += mod;
    return a;
}
 
ll subt(ll a, ll b) {
    return mode(mode(a)-mode(b));
}
 
ll add(ll a, ll b) {
    return mode(mode(a)+mode(b));
}
 
ll mult(ll a, ll b) {
    return mode(mode(a)*mode(b));
}
 
ll binpow(ll a, ll b) {
    ll res = 1;
    while (b) {
        if (b&1) res = mult(res, a);
        a = mult(a, a);
        b >>= 1;
    }
    return res;
}

const int N = 1e5+7;
vector <int> graph[N], graph2[N];
int n, subtree[N];
ll ans;

bool subtask1() {
	return sz(graph[1]) == n-1;
}

bool subtask2() {
	bool flag = 1;
	for (int i = 1; i <= n; i++) {
		if (i == 1) {
			if (sz(graph[1]) != 1 || graph[1][0] != 2) flag = 0;
		}
		else if (i == n) {
			if (sz(graph[n]) != 1 || graph[n][0] != n-1) flag = 0;
		}
		else if (sz(graph[i]) != 2 || min(graph[i][0], graph[i][1]) != i-1 || max(graph[i][0], graph[i][1]) != i+1) {
			flag = 0;
		}
	}
	return flag;
}

void dfs(int v, int p) {
	subtree[v] = 0;
	int children = 0;
	for (auto to : graph2[v]) {
		if (to == p) continue;
		children++;
		dfs(to, v);
		subtree[v] += subtree[to];
	}
	if (children == 0) subtree[v] = 1;
	if (v != p) {
		if (subtree[v]&1) ans++;
		else ans += 2;	
	}
}

int main() {
	do_not_disturb
	
	int q;
	cin >> n >> q;
	for (int i = 0; i < n-1; i++) {
		int x, y;
		cin >> x >> y;
		graph[x].pb(y);
		graph[y].pb(x);
	}
	
	if (subtask1()) {
		while (q--) {
			int m;
			cin >> m;
			vector <int> d(m), has(n+1);
			for (auto &to : d) cin >> to;
			for (auto to : d) has[to]++;
			multiset <int> st;
			ll ans = 0; 
			for (int i = 2; i <= n; i++) {
				while (has[i] > 2) {
					ans += 2;
					has[i] -= 2;
				}
				if (!has[i]) {
					st.insert(1);
				}
				else {
					while (has[i]) {
						st.insert(2);
						has[i]--;
					}
				}
			}
			if (sz(st)&1) {
				cout << -1 << endl;
			}
			else {
				while (sz(st)) {
					auto a = *st.begin();
					st.erase(st.find(a));
					auto b = *st.begin();
					st.erase(st.find(b));
					ans += a+b;
				}
				cout << ans << endl;
			}
		}
	}
	else if (subtask2()) {
		while (q--) {
			int m;
			cin >> m;
			vector <int> d(m);
			for (auto &to : d) cin >> to;
			sort(all(d));
			if (sz(d)&1) {
				cout << -1 << endl;
			}
			else {
				ll ans = n-1;
				for (int i = 0; i < sz(d); i += 2) {
					ans += abs(d[i]-d[i+1])+2;
				}
				cout << ans << endl;
			}
		}
	}
	else if (n <= 20000 && q <= 300) {
		while (q--) {
			ans = 0;
			for (int i = 1; i <= n; i++) {
				for (auto to : graph[i]) graph2[i].pb(to);
			}
			int nw = n;
			int m;
			cin >> m;
			vector <int> d(m);
			for (auto &to : d) cin >> to;
			for (auto to : d) {
				nw++;
				graph2[to].pb(nw);
				graph2[nw].pb(to);
			}
			int root = -1;
			for (int i = 1; i <= nw; i++) {
				if (sz(graph2[i]) != 1) {
					root = i;
					break;
				}
			}
			dfs(root, root);
			if (subtree[root]%2 == 0) {
				cout << ans << endl;
			}
			else {
				cout << -1 << endl;
			}
			for (int i = 1; i <= nw; i++) {
				graph2[i].clear();
			}
		}
	}
	
    return 0;
}

Subtask #1
0.0 / 0.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 4940 KB Output is correct
2 Incorrect 11 ms 5932 KB Output isn't correct

Subtask #2
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 5696 KB Output is correct
2 Correct 12 ms 5976 KB Output is correct
3 Correct 63 ms 14324 KB Output is correct
4 Correct 50 ms 13264 KB Output is correct
5 Correct 70 ms 15996 KB Output is correct

Subtask #3
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 17 ms 5536 KB Output is correct
2 Correct 17 ms 5452 KB Output is correct
3 Correct 41 ms 8148 KB Output is correct
4 Correct 53 ms 8228 KB Output is correct
5 Correct 38 ms 7796 KB Output is correct

Subtask #4
16.0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Correct 311 ms 7564 KB Output is correct
2 Correct 222 ms 6936 KB Output is correct
3 Correct 314 ms 6608 KB Output is correct
4 Correct 344 ms 7236 KB Output is correct
5 Correct 309 ms 7300 KB Output is correct
6 Correct 368 ms 7532 KB Output is correct

Subtask #5
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 28 ms 7076 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 44 ms 8224 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 30.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 4940 KB Output is correct
2 Incorrect 11 ms 5932 KB Output isn't correct