oj.uz

Submission #431913

# Submission time Handle Problem Language Result Execution time Memory
431913 2021-06-17T17:06:46 Z albertolg101 Werewolf (IOI18_werewolf) C++17
49 / 100
 1649 ms 127352 KB 
#include <bits/stdc++.h>
#include "werewolf.h"
	
using namespace std;
using pii = pair<int, int>;

struct dsu 
{
	vector<int> parent;
	vector<bool> flag;

	dsu(int n)
	{
		flag.resize(n);

		for(int i = 0 ; i < n ; i++)
			parent.push_back(i);
	}

	int Find (int nod)
	{
		if(parent[nod] == nod)
			return nod;

		return parent[nod] = Find(parent[nod]);
	}

	bool Union (int nod_a, int nod_b, bool sol)
	{
		int x = Find(nod_a),
			y = Find(nod_b);

		if(!flag[x] or !flag[y])
			return false;

		if(!sol)
			parent[min(x, y)] = max(x, y);

		if(sol)
			parent[max(x, y)] = min(x, y);

		return x != y;
	}

	void activate (int x)
	{
		flag[x] = true;
	}
};

vector<int> t;

void update (int x, int xend, int nod, int pos, int val)
{
	if(x == xend)
		return void(t[nod] = val);

	int mid = (x + xend) / 2;
	(pos <= mid ? update(x, mid, nod*2, pos, val) : update(mid+1, xend, nod*2+1, pos, val));
	
	t[nod] = max(t[nod*2], t[nod*2+1]);
}

int query (int x, int xend, int nod, int l, int r)
{
	if(x > r or xend < l)
		return 0;
	
	if(l <= x and xend <= r)
		return t[nod];

	int mid = (x + xend) / 2;
	return max(query(x, mid, nod*2, l, r), query(mid+1, xend, nod*2+1, l, r));
}
	
std::vector<int> check_validity(int n, std::vector<int> X, std::vector<int> Y,
								std::vector<int> S, std::vector<int> E,
								std::vector<int> L, std::vector<int> R) {
	
	int Q = L.size(), m = X.size();
	vector<vector<int>> g(n);
	
	for(int i = 0 ; i < m ; i++)
	{
		g[X[i]].push_back(Y[i]);
		g[Y[i]].push_back(X[i]);
	}

	dsu dsuL(n), dsuR(n);
	vector<vector<int>> gl(n), gr(n);

	for(int i = 0 ; i < n ; i++)
	{
		int nod = i;
		dsuL.activate(i);
		sort(g[i].rbegin(), g[i].rend());

		for(auto j: g[i])
		{
			int x = dsuL.Find(j);

			if(dsuL.Union(nod, x, false))
			{
				gl[nod].push_back(x);
				gl[x].push_back(nod);
				nod = dsuL.Find(nod);
			}
		}
	}

	for(int i = n - 1 ; i >= 0 ; i--)
	{
		int nod = i;
		dsuR.activate(i);
		reverse(g[i].begin(), g[i].end());

		for(auto j: g[i])
		{
			int x = dsuR.Find(j);

			if(dsuR.Union(nod, x, true))
			{
				gr[nod].push_back(x);
				gr[x].push_back(nod);
				nod = dsuR.Find(nod);
			}
		}
	}

	int cnt = 0;
	vector<int> et, levelL(n), levelR(n);
	vector<pii> indxL(n), indxR(n);
	vector<vector<int>> parentL(n, vector<int> (18)), parentR = parentL;

	function<void(int, int)> dfs1 = [&](int nod, int father)
	{
		indxR[nod].first = et.size();
		et.push_back(nod);

		for(auto i: gl[nod])
		{
			if(i != father)
			{
				levelR[i] = levelR[nod]+1;
				parentR[i][0] = nod;
				dfs1(i, nod);
			}
		}

		indxR[nod].second = et.size() - 1;
	};

	function<void(int, int)> dfs2 = [&](int nod, int father)
	{
		indxL[nod].first = cnt++;
		// cout << nod << ' ' ; 

		for(auto i: gr[nod])
		{
			if(i != father)
			{
				levelL[i] = levelL[nod] + 1;
				parentL[i][0] = nod;
				dfs2(i, nod);
			}
		}

		indxL[nod].second = cnt - 1;
	};

	// cout << endl ;
	dfs1(n-1, -1);
	// for(auto i: et)
	// 	cout << i << ' ' ; 
	// cout << endl ;
	dfs2(0, -1);
	// cout << endl << endl ;

	for(int i = 1 ; i < 18 ; i++)
	{
		for(int nod = 0 ; nod < n ; nod++)
		{
			if((1<<i) <= levelR[nod])
				parentR[nod][i] = parentR[parentR[nod][i-1]][i-1];

			if((1<<i) <= levelL[nod])
				parentL[nod][i] = parentL[parentL[nod][i-1]][i-1];
		}
	}

	vector<pair<pair<int, bool>, int>> v;

	for(int i = 0 ; i < Q ; i++)
	{
		int nod = E[i];

		for(int j = 17 ; j >= 0 ; j--)
		{
			if((1<<j) <= levelR[nod] and parentR[nod][j] <= R[i])
				nod = parentR[nod][j];
		}

		// cout << E[i] << ' ' << R[i] << ' ' << nod << ' ' << indxR[nod].first << ' ' << indxR[nod].second << endl ;

		v.push_back({{indxR[nod].first, false}, i});
		v.push_back({{indxR[nod].second, true}, i});
	}
	// cout << endl ;

	t.resize(4*n);
	sort(v.rbegin(), v.rend());

	cnt = 0;
	vector<int> mm(Q), vans;
	vector<pii> ans;

	// cout << endl ;
	// for(int i = v.size() - 1 ; i >= 0 ; i--)
	// 	cout << v[i].first.first << ' ' << v[i].first.second << ' ' << v[i].second << endl ;
	// cout << endl ;

	for(int i = 0 ; i < n ; i++)
	{
		while(v.size() and v.back().first.first == i and !v.back().first.second)
		{
			mm[v.back().second] = ++cnt;
			v.pop_back();
		}

		// cout << "u " << et[i] << ' ' << indxL[et[i]].first << ' ' << cnt << endl ;
		update(1, n, 1, indxL[et[i]].first, cnt);

		while(v.size() and v.back().first.first == i and v.back().first.second)
		{
			int nod = S[v.back().second];

			for(int j = 17 ; j >= 0 ; j--)
			{
				if((1<<j) <= levelL[nod] and parentL[nod][j] >= L[v.back().second])
					nod = parentL[nod][j];
			}

			// cout << nod << ' ' << L[v.back().second] << ' ' << indxL[nod].first << ' ' << indxL[nod].second << ' ' << query(1, n, 1, indxL[nod].first, indxL[nod].second) << ' ' << mm[v.back().second] << endl ;

			ans.push_back({v.back().second, query(1, n, 1, indxL[nod].first, indxL[nod].second) >= mm[v.back().second]});
			v.pop_back();
		}
	}

	sort(ans.begin(), ans.end());
	for(auto i: ans)
		vans.push_back(i.second);

	return vans;
}

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 332 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 332 KB Output is correct
6 Correct 1 ms 384 KB Output is correct
7 Correct 1 ms 332 KB Output is correct
8 Correct 1 ms 332 KB Output is correct
9 Correct 1 ms 332 KB Output is correct

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 332 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 332 KB Output is correct
6 Correct 1 ms 384 KB Output is correct
7 Correct 1 ms 332 KB Output is correct
8 Correct 1 ms 332 KB Output is correct
9 Correct 1 ms 332 KB Output is correct
10 Correct 9 ms 1992 KB Output is correct
11 Correct 9 ms 1868 KB Output is correct
12 Correct 9 ms 1804 KB Output is correct
13 Correct 9 ms 2124 KB Output is correct
14 Correct 8 ms 2104 KB Output is correct
15 Correct 12 ms 1996 KB Output is correct

Subtask #3
34.0 / 34.0

# Verdict Execution time Memory Grader output
1 Correct 1359 ms 102640 KB Output is correct
2 Correct 1438 ms 116760 KB Output is correct
3 Correct 1249 ms 112984 KB Output is correct
4 Correct 1216 ms 111412 KB Output is correct
5 Correct 1157 ms 111360 KB Output is correct
6 Correct 1346 ms 111124 KB Output is correct
7 Correct 1277 ms 111000 KB Output is correct
8 Correct 1311 ms 116696 KB Output is correct
9 Correct 1036 ms 112932 KB Output is correct
10 Correct 915 ms 111400 KB Output is correct
11 Correct 907 ms 111240 KB Output is correct
12 Correct 1014 ms 111108 KB Output is correct
13 Correct 1488 ms 120408 KB Output is correct
14 Correct 1480 ms 120432 KB Output is correct
15 Correct 1546 ms 120432 KB Output is correct
16 Correct 1454 ms 120408 KB Output is correct
17 Correct 1289 ms 111012 KB Output is correct

Subtask #4
0 / 51.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 332 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 332 KB Output is correct
6 Correct 1 ms 384 KB Output is correct
7 Correct 1 ms 332 KB Output is correct
8 Correct 1 ms 332 KB Output is correct
9 Correct 1 ms 332 KB Output is correct
10 Correct 9 ms 1992 KB Output is correct
11 Correct 9 ms 1868 KB Output is correct
12 Correct 9 ms 1804 KB Output is correct
13 Correct 9 ms 2124 KB Output is correct
14 Correct 8 ms 2104 KB Output is correct
15 Correct 12 ms 1996 KB Output is correct
16 Correct 1359 ms 102640 KB Output is correct
17 Correct 1438 ms 116760 KB Output is correct
18 Correct 1249 ms 112984 KB Output is correct
19 Correct 1216 ms 111412 KB Output is correct
20 Correct 1157 ms 111360 KB Output is correct
21 Correct 1346 ms 111124 KB Output is correct
22 Correct 1277 ms 111000 KB Output is correct
23 Correct 1311 ms 116696 KB Output is correct
24 Correct 1036 ms 112932 KB Output is correct
25 Correct 915 ms 111400 KB Output is correct
26 Correct 907 ms 111240 KB Output is correct
27 Correct 1014 ms 111108 KB Output is correct
28 Correct 1488 ms 120408 KB Output is correct
29 Correct 1480 ms 120432 KB Output is correct
30 Correct 1546 ms 120432 KB Output is correct
31 Correct 1454 ms 120408 KB Output is correct
32 Correct 1289 ms 111012 KB Output is correct
33 Correct 1461 ms 112924 KB Output is correct
34 Correct 505 ms 40208 KB Output is correct
35 Correct 1649 ms 117508 KB Output is correct
36 Correct 1397 ms 112152 KB Output is correct
37 Correct 1553 ms 116284 KB Output is correct
38 Correct 1550 ms 113332 KB Output is correct
39 Correct 1310 ms 127352 KB Output is correct
40 Correct 1507 ms 122504 KB Output is correct
41 Correct 1236 ms 115364 KB Output is correct
42 Incorrect 1081 ms 112160 KB Output isn't correct
43 Halted 0 ms 0 KB -