CodeForces - 1471F-Strange Housing（ bfs + 贪心 ）

题目链接：点击进入

题目

题意

n个房间，m条路，选一些房间让老师来住，要求
1、如果两个房间中都没有老师，它们之间的路将关闭
2、整幅图为n点的连通图
3、两个老师的房间不能直接相连，只能间接相连
如果存在这样的图，则输出YES以及老师的房间的个数和房间号，否则输出NO
（ 输出任意一种方案即可 ）

思路

任意选一个点作为 bfs 起点
如果这个点所有相邻的点都没有被选，则选择这个点当作老师的房间；
如果这个点存在一个相邻的点被选了，则不选这个点当作老师的房间；
如果最后是一个大小为 n 的连通图，那么就输出 YES，同时输出标记为被选择的点以及个数
如果不是，那么输出 NO

代码

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define max(a,b) a>b?a:b
//#define min(a,b) a<b?a:b
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e6+10;
const int mod=1e9+7;
const double eps=1e-9;
const int N=1e6+10;
int t,n,m,ans,sum;
int head[maxn],tot;
bool vis[maxn],fin[maxn];
struct node
{
	int to;
	int next;
}edge[maxn];
void init()
{
	for(int i=0;i<=n;i++)
		head[i]=-1,vis[i]=0,fin[i]=0;	
	tot=0;
}
void add(int u,int v)
{
	edge[tot].to=v;
	edge[tot].next=head[u];
	head[u]=tot++;
}
void bfs()
{
	queue<int>q;
	ans=1;
	sum=0;
	vis[1]=1;
	q.push(1);
	while(q.size())
	{
		int now=q.front(),flag=0;
		q.pop();
		for(int i=head[now];i!=-1;i=edge[i].next)
		{
			int v=edge[i].to;
			if(fin[v]==1) flag=1;
			if(!vis[v])
			{
				ans++;
				vis[v]=1;
				q.push(v);
			}
		}
		if(!flag) fin[now]=1,sum++;
	}
	if(ans==n)
	{
		cout<<"YES"<<endl;
		cout<<sum<<endl;
		for(int i=1;i<=n;i++)
			if(fin[i]==1)
				cout<<i<<" ";
		cout<<endl;
	}
	else
		cout<<"NO"<<endl;
}
int main()
{	
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);	
	cin>>t;
	while(t--)
	{
		cin>>n>>m;init();
		for(int i=1;i<=m;i++)
		{
			int x,y;
			cin>>x>>y;
			add(x,y);
			add(y,x);
		}
		bfs();
	}
    return 0;
}