codeforces723F st-Spanning Tree(连通性t)

题意：

给出一个n个顶点m条边的无向图，并给出两个点s，t和对应的度数，要求将图转变为一颗生成树，并且s，t的度数要满足要求。

要点：

生成树也就是把原图的边减少至n-1条边，并且所有的点都连通。这题的思路是将st除去的图进行连通操作，然后此时这些连通块可以分为二种：

1.只和st其中一个相连

2.和st两个都相连

我们需要先将第一种连通块的先连在st上，因为它是必须要连的，否则没有地方可以连。接着连接第二种连通块与st连接，这一步中，只要一个连通块与st均连接，其他的与st其中任意一个连接。这里一开始我觉得很难写，后来发现只要第一个连通块与st均连接，st就已经连接，之后再进行连接操作，连通块也不会同时连接到st两个上。最后考虑一下如果到最后还不能使st相连，只能连接原本存在的st这条边。原则上只要第二种连通块存在就不连接原本的st边，因为第二种连通块本身起码要要连接st中一个，就算两个都连，对度数消耗也就是2，而如果连接st原边，度数消耗就是3。

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 400000 + 5;
int n, m;
pair<int, int> edge[maxn];
int used[maxn], fa[maxn],dd[maxn];
int cs[maxn], ct[maxn];

int find(int x)//并查集
{
	return fa[x] == x ? x : (fa[x] = find(fa[x]));
}

int main()
{
	int i, j;
	int s, t, ds, dt;
	int st = 0, cnt = 0;
	scanf("%d%d", &n, &m);
	for (i = 1; i <= m; i++)
		scanf("%d%d", &edge[i].first, &edge[i].second);
	scanf("%d%d%d%d", &s, &t, &ds, &dt);
	for (i = 1; i <= m; i++)
		if (edge[i] == make_pair(s, t) || edge[i] == make_pair(t, s))
			st = i;			//s与t这条边是否存在
	for (i = 1; i <= n; i++)
		fa[i] = i;
	for (i = 1; i <= m; i++)//先将除st以外的点连通起来
	{
		int u = edge[i].first, v = edge[i].second;
		if (u == s || v == s || u == t || v == t)
			continue;
		dd[i] = 1;
		if (find(u) == find(v))
			continue;
		fa[find(u)] = find(v);
		cnt++;					
		used[i] = 1;
	}
	for (i = 1; i <= m; i++)//这步是判断每个连通块是否与st连接
	{
		if (dd[i]) continue;
		if (st == i) continue;
		int u = edge[i].first, v = edge[i].second;
		if (u == s) cs[find(v)] = 1;
		if (u == t) ct[find(v)] = 1;
		if (v == s) cs[find(u)] = 1;
		if (v == t) ct[find(u)] = 1;
	}
	for (i = 1; i <= m; i++)//这一步就是将只有与st其中一个点连通的连通块和st连通
	{
		if (dd[i]) continue;
		if (st == i) continue;
		int u = edge[i].first, v = edge[i].second;
		if (!ds && (u == s || v == s)) continue;
		if (!dt && (u == t || v == t)) continue;
		if (u != s&&u != t && (cs[find(u)]&& ct[find(u)])) continue;
		if (v != s&&v != t && (cs[find(v)]&& ct[find(v)])) continue;
		if (find(u) == find(v))
			continue;
		if (u == s || v == s) ds--;
		if (u == t || v == t) dt--;
		fa[find(u)] = find(v);
		cnt++;
		used[i] = 1;
	}
	for (i = 1; i <= m; i++)//这一步就是处理与st两个都连通的连通块
	{						//注意只要第一个连通块均与st相连，st就已经连通，后面的连通块只能加到st其中一个上
		if (dd[i]) continue;
		if (st == i) continue;
		int u = edge[i].first, v = edge[i].second;
		if (!ds && (u == s || v == s)) continue;
		if (!dt && (u == t || v == t)) continue;
		if (find(u) == find(v))
			continue;
		if (u == s || v == s) ds--;
		if (u == t || v == t) dt--;
		fa[find(u)] = find(v);
		cnt++;
		used[i] = 1;
	}
	if (find(s) != find(t) && ds&&dt)//如果到最后st都不连通，说明没有和st都连接的连通块，只能连通st这条边
	{
		used[st] = 1;
		cnt++;
	}
	if (cnt < n - 1)
	{
		printf("No\n");
		return 0;
	}
	else
	{
		printf("Yes\n");
		for (i = 1; i <= m; i++)
			if (used[i])
				printf("%d %d\n", edge[i].first, edge[i].second);
	}
	return 0;
}