[补题记录] Atcoder Beginner Contest 299(E)

URL:https://atcoder.jp/contests/abc299

[补题记录] Atcoder Beginner Contest 299(E)_第1张图片

目录

E

Problem/题意

Thought/思路

Code/代码


E

Problem/题意

给出 N(1 <= N <= 2000)个点和 M 条边的一个无向图,要求用白色和黑色对这个图染色。

满足下面两个条件,则输出 Yes 和各点的颜色,否则输出 No:

  • 至少有一个黑色点;
  • 给出 K 个条件(p,d),要求距离 p 最短距离为 d 的点至少一个需要染成黑色;

Thought/思路

这道题的坑点在于,题目没有很清楚地说明,只要有一个黑色点与 p 最短句路为 d 即可。因此很容易误认为所有距离 p 为 d 的点都要染黑。

搞清楚这点思路就很明确了,对于每个要求(p,d),bfs 把所有最短距离小于 d 的点都染成白色;

再次 bfs,判断是否有一个距离为 d 的且不为白色的点,有则 return true,否则这个要求不能满足,输出 No。

Code/代码

#include "bits/stdc++.h"

int n, m, dis[2003], ans[2003];
std::vector  g[2003];

struct K {
	int p, d;
}tmp[2003];

void white(int s, int d) {
	for (int i = 1; i <= n; ++ i) dis[i] = INT_MAX;

	std::queue > q;
	q.push({s, 0});
	dis[s] = 0;

	while (!q.empty()) {
		auto o = q.front(); q.pop();

		int x = o.first, pre = o.second;

		for (auto &next : g[x]) {
			if (next == pre) continue;
			if (dis[next] > dis[x] + 1) {
				dis[next] = dis[x] + 1;
				q.push({next, x});
			}
		}
	}

	for (int i = 1; i <= n; ++ i) {
		if (dis[i] < d) ans[i] = 0;
	}
}

bool black(int s, int d) {
	for (int i = 1; i <= n; ++ i) dis[i] = INT_MAX;

	std::queue > q;
	q.push({s, 0});
	dis[s] = 0;

	while (!q.empty()) {
		auto o = q.front(); q.pop();

		int x = o.first, pre = o.second;

		for (auto &next : g[x]) {
			if (next == pre) continue;
			if (dis[next] > dis[x] + 1) {
				dis[next] = dis[x] + 1;
				q.push({next, x});
			}
		}
	}

	for (int i = 1; i <= n; ++ i) {
		if (dis[i] == d and ans[i] != 0) {
			ans[i] = 1;
			return true;
		}
	}

	return false;
}

signed main () {
	std::cin >> n >> m;
	for (int i = 1; i <= m; ++ i) {
		int x, y; std::cin >> x >> y;
		g[x].push_back(y);
		g[y].push_back(x);
	}

	for (int i = 1; i <= n; ++ i) ans[i] = -1;

	int k; std::cin >> k;
	for (int i = 1; i <= k; ++ i) {
		std::cin >> tmp[i].p >> tmp[i].d; // 题目保证:pi < pj [i < j]
		white(tmp[i].p, tmp[i].d);
	}


	bool flag = true;
	for (int i = 1; i <= k; ++ i) {
		int s = tmp[i].p, d = tmp[i].d;
		if (!black(s, d)) {
			flag = false;
			break;
		}
	}

	if (!flag) {
		std::cout << "No";
	} else {
		if (k == 0) ans[1] = 1;
		std::cout << "Yes\n";
		for (int i = 1; i <= n; ++ i) 
			std::cout << (ans[i] == -1 ? 0 : ans[i]);
	}

	return 0;
}

你可能感兴趣的:(补题记录,java,算法,数据结构)