oj.uz

Submission #404937

# Submission time Handle Problem Language Result Execution time Memory
404937 2021-05-15T10:31:44 Z silverfish Spring cleaning (CEOI20_cleaning) C++14
19 / 100
 1000 ms 13772 KB 
#include <bits/stdc++.h>
using namespace std;

/*<DEBUG>*/
#define tem template <typename 
#define can_shift(_X_, ...) enable_if_t<sizeof test<_X_>(0) __VA_ARGS__ 8, debug&> operator<<(T i)
#define _op debug& operator<<
tem C > auto test(C *x) -> decltype(cerr << *x, 0LL);
tem C > char test(...);
tem C > struct itr{C begin, end; };
tem C > itr<C> get_range(C b, C e) { return itr<C>{b, e}; }
struct debug{
#ifdef _LOCAL
	~debug(){ cerr << endl; }
	tem T > can_shift(T, ==){ cerr << boolalpha << i; return *this; }
	tem T> can_shift(T, !=){ return *this << get_range(begin(i), end(i)); }
	tem T, typename U > _op (pair<T, U> i){ 
		return *this << "< " << i.first << " , " << i.second << " >"; }
	tem T> _op (itr<T> i){
		*this <<  "{ ";
		for(auto it = i.begin; it != i.end; it++){
			*this << " , " + (it==i.begin?2:0) << *it;
		}
		return *this << " }";
	}
#else
tem T> _op (const T&) { return *this; }
#endif 
};

tem T>
string _ARR_(T* arr, int sz){
	string ret = "{ " + to_string(arr[0]); 
	for(int i = 1; i < sz; i++) ret += " , " +  to_string(arr[i]);
	ret += " }"; return ret;
}

#define exp(...) " [ " << #__VA_ARGS__ << " : " << (__VA_ARGS__) << " ]"
/*</DEBUG>*/

typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef unsigned int uint;
typedef pair<int, int> pii;
//mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());

#define pb push_back
#define FAST ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define TC int __TC__; cin >> __TC__; while(__TC__--)
#define ar array

const int INF = 1e9 + 7, N = 2e5 + 10;

vector<int> adj[N];
int win[N], wout[N], in[N], out[N], add[N];
int ans = 0, n, q, d, cur,fst=0, leafs = 0, par[N], nin, nout;
bool was_leaf[N], leaf[N];

void dfs(int u, int p=fst){
	par[u] = p;
	for(int v : adj[u]){
		if(v != p){
			dfs(v, u);
			in[u] += out[v];
		}
	}


	if(adj[u].size() == 1){
		in[u] = 0;
		out[u] = 1;
		++leafs;
		leaf[u] = 1;
	}else{
		out[u] = (in[u]%2 ? 1 : 2);
	}

	ans += in[u];
	win[u] = in[u];
	wout[u] = out[u];
	return;
}

int main()
{
	FAST;
	cin >> n >> q;
	for(int i = 0; i < n-1; ++i){
		int u, v; cin >> u >> v;
		--u; --v;
		adj[u].pb(v);
		adj[v].pb(u);
	}

	for(int i = 0; i < n; ++i){
		if(adj[fst].size() == 1 && adj[i].size() != 1) fst = i;
	}

	dfs(fst);
	debug() << exp(ans);

	while(q--){
		cur = n-1;
		cin >> d;

		for(int i = 0; i < d; ++i){
			++cur;
			cin >> add[i];
			--add[i];
			if(leaf[add[i]]){
				was_leaf[i] = 1;
				leaf[add[i]] = 0;
				--leafs;
			}
			++leafs;

			adj[add[i]].pb(cur);
			in[cur] = 0;
			out[cur] = 1;

			int u = add[i];

			while(1){
				ans -= in[u];
				in[u] = out[u] = 0;
				for(int v : adj[u]){
					if(v != par[u]){
						in[u] += out[v];
					}
				}

				out[u] = (in[u]%2 ? 1 : 2);
				ans += in[u];
				if(u == fst) break;
				u = par[u];
			}
		}

		debug() << exp(ans);
		cout << (leafs%2 ? -1 : ans) << '\n';

		for(int i = 0; i < d; ++i){
			adj[add[i]].pop_back();
			if(was_leaf[i]){
				leafs++;
				leaf[add[i]] = 1;
				was_leaf[i] = 0;
			} 
			leafs--;

			int u = add[i];
			while(1){
				ans += win[u] - in[u];
				in[u] = win[u];
				out[u] = wout[u];
				if(u == fst) break;
				u = par[u];
			}
		}

		debug() << exp(ans);
	}


	return 0;
}

Subtask #1
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4940 KB Output is correct
2 Correct 83 ms 6224 KB Output is correct

Subtask #2
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 250 ms 6548 KB Output is correct
2 Correct 249 ms 6624 KB Output is correct
3 Correct 750 ms 10756 KB Output is correct
4 Execution timed out 1096 ms 9272 KB Time limit exceeded
5 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

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

Subtask #4
0 / 16.0

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

Subtask #5
19.0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 82 ms 8616 KB Output is correct
2 Correct 130 ms 8700 KB Output is correct
3 Correct 96 ms 7820 KB Output is correct
4 Correct 111 ms 9924 KB Output is correct
5 Correct 107 ms 9956 KB Output is correct

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 202 ms 10996 KB Output is correct
2 Execution timed out 1031 ms 13772 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #7
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4940 KB Output is correct
2 Correct 83 ms 6224 KB Output is correct
3 Correct 250 ms 6548 KB Output is correct
4 Correct 249 ms 6624 KB Output is correct
5 Correct 750 ms 10756 KB Output is correct
6 Execution timed out 1096 ms 9272 KB Time limit exceeded
7 Halted 0 ms 0 KB -