답안 #530221

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
530221 2022-02-24T20:01:26 Z luciocf Colors (RMI18_colors) C++14
7 / 100
1260 ms 26068 KB
#include <bits/stdc++.h>

#define ff first
#define ss second

using namespace std;

typedef pair<int, int> pii;

const int maxn = 5e5+10;
const int inf = 1e9+10;

int n, m;
int a[maxn], b[maxn];
pii p[maxn];

vector<int> quem[maxn];

bool mark[maxn];

vector<int> grafo[maxn];

struct BinaryLifting
{
	int nivel[maxn];
	int tab[maxn][20], mn[maxn][20], mx[maxn][20];

	void dfs(int u, int p)
	{
		for (auto v: grafo[u])
		{
			if (v == p) continue;

			nivel[v] = nivel[u]+1, tab[v][0] = u;
			dfs(v, u);
		}
	}

	void build(void)
	{
		for (int i = 1; i <= n; i++)
		{
			if (tab[i][0] == 0) mx[i][0] = b[i];
			else mx[i][0] = max(b[i], b[tab[i][0]]);
			
			if (tab[i][0] == 0) mn[i][0] = a[i];
			else mx[i][0] = min(a[i], a[tab[i][0]]);
		}

		for (int j = 1; j < 20; j++)
		{
			for (int i = 1; i <= n; i++)
			{
				tab[i][j] = tab[tab[i][j-1]][j-1];
				mx[i][j] = max(mx[i][j-1], mx[tab[i][j-1]][j-1]);
				mn[i][j] = min(mn[i][j-1], mn[tab[i][j-1]][j-1]);
			}
		}
	}

	int get_mn(int u, int v)
	{
		if (nivel[u] < nivel[v]) swap(u, v);

		int ans = inf;

		for (int i = 19; i >= 0; i--)
			if (nivel[u]-(1<<i) >= nivel[v])
				ans = min(ans, mn[u][i]), u = tab[u][i];

		if (u == v) return ans;

		for (int i = 19; i >= 0; i--)
			if (tab[u][i] && tab[v][i] && tab[u][i] != tab[v][i])
				ans = min({ans, mn[u][i], mn[v][i]}), u = tab[u][i], v = tab[v][i];

		return min({ans, mn[u][0], mn[v][0]});
	}

	int get_mx(int u, int v)
	{
		if (nivel[u] < nivel[v]) swap(u, v);

		int ans = 0;

		for (int i = 19; i >= 0; i--)
			if (nivel[u]-(1<<i) >= nivel[v])
				ans = max(ans, mx[u][i]), u = tab[u][i];

		if (u == v) return ans;

		for (int i = 19; i >= 0; i--)
			if (tab[u][i] && tab[v][i] && tab[u][i] != tab[v][i])
				ans = max({ans, mx[u][i], mx[v][i]}), u = tab[u][i], v = tab[v][i];

		return max({ans, mx[u][0], mx[v][0]});
	}
} LCA;

void dfs(int u, int c)
{
	mark[u] = 1;

	for (auto v: grafo[u])
		if (!mark[v] && a[v] >= c && b[v] <= c)
			dfs(v, c);
}

bool reach(int u, int v, int c)
{
	memset(mark, 0, sizeof mark);
	dfs(u, c);

	return mark[v];
}

void solve_tree(void)
{
	bool ok = 1;

	for (int i = 1; i <= n; i++)
		if (a[i] < b[i])
			ok = 0;

	LCA.dfs(1, 0);
	LCA.build();

	for (int u = 1; u <= n; u++)
	{
		int c = b[u], v = quem[c][0];

		if (LCA.get_mn(u, v) < c && LCA.get_mx(u, v) > c)
			ok = 0;
	}

	printf("%d\n", ok);
}

int main(void)
{
	int tc;
	scanf("%d", &tc);

	while (tc--)
	{
		scanf("%d %d", &n, &m);

		vector<int> bla;

		for (int i = 1; i <= n; i++)
		{
			grafo[i].clear();
			quem[i].clear();
		}

		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &a[i]);

			bla.push_back(a[i]);
			quem[a[i]].push_back(i);
		}

		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &b[i]);

			p[i] = {b[i], i};
		}

		for (int i = 1; i <= m; i++)
		{
			int u, v;
			scanf("%d %d", &u, &v);

			grafo[u].push_back(v);
			grafo[v].push_back(u);
		}

		sort(bla.begin(), bla.end());

		bool sub_tree = (m == n-1);
		for (int i = 0; i < bla.size(); i++)
			if (bla[i] != i+1)
				sub_tree = 0;

		if (sub_tree)
		{
			solve_tree();
			continue;
		}

		sort(p+1, p+n+1);

		bool ok = 1;

		for (int i = n; i >= 1; i--)
		{
			int c = p[i].ff, u = p[i].ss;

			bool flag = 0;

			for (auto v: quem[c])
				if (reach(v, u, c))
					flag = 1;

			ok &= flag;
		}

		for (int i = 1; i <= n; i++)
			if (a[i] < b[i])
				ok = 0;

		if (ok) printf("1\n");
		else printf("0\n");
	}
}

Compilation message

colors.cpp: In function 'int main()':
colors.cpp:183:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  183 |   for (int i = 0; i < bla.size(); i++)
      |                   ~~^~~~~~~~~~~~
colors.cpp:142:7: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  142 |  scanf("%d", &tc);
      |  ~~~~~^~~~~~~~~~~
colors.cpp:146:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  146 |   scanf("%d %d", &n, &m);
      |   ~~~~~^~~~~~~~~~~~~~~~~
colors.cpp:158:9: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  158 |    scanf("%d", &a[i]);
      |    ~~~~~^~~~~~~~~~~~~
colors.cpp:166:9: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  166 |    scanf("%d", &b[i]);
      |    ~~~~~^~~~~~~~~~~~~
colors.cpp:174:9: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  174 |    scanf("%d %d", &u, &v);
      |    ~~~~~^~~~~~~~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Incorrect 1216 ms 25848 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 382 ms 26004 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Incorrect 1260 ms 25832 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 1260 ms 25832 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 1216 ms 25848 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 159 ms 26068 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 319 ms 25072 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 1216 ms 25848 KB Output isn't correct
2 Halted 0 ms 0 KB -