Codeforces Round #375 (Div. 2)F. st-Spanning Tree

题目链接：F. st-Spanning Tree

题解：先把与s,t无关的边遍历一次，把能够链接两个不同的并查集连起来，然后在把每个块标记一下看能否与s,t相连，如果有一个块s和t都可以连就先把s,t联通，然后把只能与s和只能t相连的块分别与s,t相连

连的过程中维护ds,dt,最后在贪心把两个都可以连的连上去.最后看连的边数是否有n-1条，有的话就yes输出答案

By lhclqslove, contest: Codeforces Round #375 (Div. 2), problem: (F) st-Spanning Tree, Accepted, #
 #include<bits/stdc++.h>
#include<set>
#include<cstdio>
#include<iomanip>
#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#define pb push_back
#define ll long long
#define fi first
#define se second
#define PI 3.14159265
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define eps 1e-7
#define pii pair<int,int>
typedef unsigned long long ull;
const int mod=1e3+5;
const ll inf=0x3f3f3f3f3f3f3f;
const int maxn=2e5+5;
using namespace std;
int n,m,f[maxn],x[maxn*2],y[maxn*2],link1[maxn],link2[maxn];
vector<pii>edge;
set<int>yu;
bool vis[maxn];
vector<pii>ans;
set<int>ss,tt;
void init()
{
    for(int i=0;i<maxn;i++)f[i]=i;
}
int find(int x)
{
    if(f[x]==x)return x;
    else return f[x]=find(f[x]);
}
void link(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    f[fy]=fx;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    init();
    for(int i=0;i<m;i++)
    {
        cin>>x[i]>>y[i];
    }
    int s,t,ds,dt;
    cin>>s>>t>>ds>>dt;
    if(m<n-1)
    {
        cout<<"No"<<endl;return 0;
    }
    for(int i=0;i<m;i++)
    {
        if(x[i]!=s&&x[i]!=t&&y[i]!=t&&y[i]!=s)
        {
            edge.pb({x[i],y[i]});
        }
        else
        {
            if(x[i]==s)ss.insert(y[i]);
            if(y[i]==s)ss.insert(x[i]);
            if(x[i]==t)tt.insert(y[i]);
            if(y[i]==t)tt.insert(x[i]);
        }
    }
    for(int i=0;i<edge.size();i++)
    {
        if(find(edge[i].fi)!=find(edge[i].se))
        {
            ans.pb(edge[i]);
            link(edge[i].se,edge[i].fi);
        }
    }
    for(int to:tt)
    {
        int fto=find(to);
        link1[fto]=1;
    }
     for(int to:ss)
    {
        int fto=find(to);
        link2[fto]=1;
    }bool flag=false;
    if(!flag)//st相连
    {
        for(int to:tt)
        {
            if(to==s)continue;
            int fto=find(to);
            if(link2[fto])
            {//cout<<to<<"UUU"<<t<<endl;
                if(dt){ans.pb({t,to});dt--;link(to,t);vis[find(to)]=1;break;}
            }
        }
        for(int to:ss)
        {
            if(to==t)continue;
            if(vis[find(to)])
            {
                //cout<<to<<"SSS"<<s<<endl;
                if(ds)ans.pb({s,to}),ds--,link(to,s);
                flag=true;
                break;
            }
        }
    }
   // cout<<ds<<" "<<dt<<endl;
    for(int to:tt)//只与t相连
    {
        if(to==s)continue;
        int fto=find(to);
        if(!link2[fto]&&fto!=find(t))
        {
           // cout<<"####"<<endl;
            if(dt)ans.pb({to,t}),dt--,link(to,t);
            else
            {
                cout<<"No"<<endl;return 0;
            }
        }
    }
    for(int to:ss)//只与s相连
    {
        if(to==t)continue;
        int fto=find(to);
        if(!link1[fto]&&fto!=find(s))
        {
          //  cout<<"$$$$"<<endl;
            if(ds)ans.pb({to,s}),ds--,link(s,to);
            else
            {
                cout<<"No"<<endl;return 0;
            }
        }
    }
    for(int to:ss)//个都相连
    {
        if(to==t)continue;
        int fto=find(to);
        if(fto!=find(s))
        {
            if(ds)
            {
                ans.pb({to,s});ds--,link(s,to);
            }
        }
    }
    for(int to:tt)
    {
        if(to==s)continue;
        int fto=find(to);
        if(fto!=find(t))
        {
            if(dt)
            {
                ans.pb({to,t});dt--,link(to,t);
            }
        }
    }
    if(find(s)!=find(t))
    {
        if(ds&&dt&&ss.count(t))
        {
            ans.pb({s,t});ds--;dt--;
        }
        else
        {
            cout<<"No"<<endl;return 0;
        }
    }
    if(ans.size()==n-1)
    {
        cout<<"Yes"<<endl;
        for(pii tmp :ans)
        {
            cout<<tmp.fi<<" "<<tmp.se<<endl;
        }
    }
    else
    {
        cout<<"No"<<endl;
    }
    return 0;
}