这道题还是非常有意思的,题意很简单,就是给定一个图,和图上的双向边,要求1号节点的度(连接边的条数)等于K,求这棵树的生成树。
我们首先要解决,如何让1号节点的度时为k的呢???而且求的是生成树,意思是不是所有边都会选择。那么我们如何选择才能保证1号节点有K个度呢???这里就要考虑联通分量的问题了,我们刨除1号点,那么联通分量的个数,就是我们让图联通的最小个数,因此我们需要用并查集,把点分在不同的联通块内部。
再考虑我们每个联通块,至少需要1条连接1号点的边。不够K再添加连接1号点的边。
然后考虑由于是生成树,我们可以枚举每个联通块连接1的点,DFS找出生成树边,记录即可。
DFS+并查集版本:
#include<iostream> #include<string.h> #include<stdio.h> #include<algorithm> #include<vector> #define pii pair<int,int> #define mp make_pair using namespace std; const int maxx = 2e5+7; int fa[maxx]; vector<int>G[maxx]; vector<int>p; vector<pii>ans; struct node { int u,v; } a[maxx]; int vis[maxx]; int v[maxx]; int Find(int x) { return fa[x]==x?x:(fa[x]=Find(fa[x])); } void dfs(int x) { int nex; for (int i=0; i<G[x].size(); i++) { nex=G[x][i]; if (nex==1)continue; if (v[nex]==0) { v[nex]=1; ans.push_back(mp(x,nex)); dfs(nex); } } } int main() { int n,m,k; while(~scanf("%d%d%d",&n,&m,&k)) { ans.clear(); memset(vis,0,sizeof(vis)); memset(v,