Codeforces Round #694 (Div. 2) F Strange Housing

题意: 有n个房间，m条通道，其中的一些房间中会居住着教师，教师的房间有三点要求
1: 两个教师房间不能由一条通道直接相连，只能间接相连
2: 如果两个房间中都没有教师，它们之间的直接通道将被关闭
3: 在关闭掉2条件中的通道之后，整幅图为连通图(包括有教师的房间和无教师的房间)
如果存在这样的图则输出YES和居住教师的房间的个数和房间号，否则输出NO
思路: 一开始没有看清题目，以为要求找到可居住的房间的个数的最大值，一直没想到复杂度足够低的方法，实际上只需要输出任意方案。
很容易可以发现，只要图是联通的，就一定存在答案，否则就NO。
我们将有教师的房间标记为1，无教师的房间标记为2，因为两个教师的房间不能直接相连，所以如果一个房间被标记为1，与其相连的其他房间都标记为2。
从任意一个点出发，将其放入队列中，标记为2。然后将与其相连的未标记的点标记未1，从该点开始进行bfs，与该点相连的点全标记为2然后放进队列。
队列部分代码:

		q.push(1);
		vis[1] = 2;
		while (!q.empty()) {
			int p = q.front();
			q.pop();
			for (int i = head[p]; ~i; i = edge[i].next) {
				int v = edge[i].to;
				if (!vis[v]) {
					ans++;//记录走过的房间的个数
					bfs(v);
				}
			}
		}

bfs部分代码

void bfs(int fa) {
	vis[fa] = 1;
	ans1++;//记录可居住的房间的个数
	for (int i = head[fa]; ~i; i = edge[i].next) {
		int v = edge[i].to;
		if (!vis[v]) {
			ans++;//记录走过的点的个数
			vis[v] = 2;
			q.push(v);
		}
	}
}

如果ans==n，则说明图是联通图，输出相应的答案。
否则图不连通，输出NO。
完整代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<stack>
#define ll long long 
using namespace std;
const int inf = 1e9 + 7;
struct Edge {
	int to, next;
};
int ans = 1, ans1 = 1;
Edge edge[700005];
queue<int>q;
int head[300005], vis[300005];
int tot = 0;
void addedge(int u, int v) {
	edge[tot].to = v;
	edge[tot].next = head[u];
	head[u] = tot++;
}
void bfs(int fa) {
	vis[fa] = 1;
	ans1++;//记录可居住的房间的个数
	for (int i = head[fa]; ~i; i = edge[i].next) {
		int v = edge[i].to;
		if (!vis[v]) {
			ans++;//记录走过的点的个数
			vis[v] = 2;
			q.push(v);
		}
	}
}
int main()
{
	int t;
	scanf("%d", &t);
	while (t--) {
		ans = 1, ans1 = 0,tot=0;
		int n, m;
		scanf("%d%d", &n, &m);
		for (int i = 1; i <= n; i++) { head[i] = -1, vis[i] = 0; }
		tot = 0;
		int u, v;
		for (int i = 1; i <= m; i++) {
			scanf("%d%d", &u, &v);
			addedge(u, v);
			addedge(v, u);
		}
		q.push(1);
		vis[1] = 2;
		while (!q.empty()) {
			int p = q.front();
			q.pop();
			for (int i = head[p]; ~i; i = edge[i].next) {
				int v = edge[i].to;
				if (!vis[v]) {
					ans++;//记录走过的房间的个数
					bfs(v);
				}
			}
		}
		if (ans == n) {
			printf("YES\n");
			printf("%d\n", ans1);
			for (int i = 1; i <= n; i++) {
				if (vis[i] == 1)printf("%d ", i);
			}
			printf("\n");
		}
		else printf("NO\n");
	}
	return 0;
}

PS (据说用memset会T，所以在这不推荐使用memset)