E-Directing Edges(拓扑排序)

题目传送门

题意: 给你一张n个点m条边的图,其中有一些边是有向边,有一些边是无向边,题目要求你对所有无向边选择一个方向,使得整个图成为有向无环图(DAG),若无法做到则输出-1。

思路: 如果给定的有向边已经形成环了,那么再怎么改无向边,都无法做到。如果有向边没有形成环,那么就可以做到。我们把有向边连接起来,无向边不连接(看做一个个孤立的点),对整张图进行拓扑排序,因为每个点只有1次入队出队的机会,所以我们可以得到每个点出队的顺序。我们把每条边按照这个顺序输出就行。(无向边迎合有向边的方向,有向边中,起点肯定比终点早出队,所以我们只要把无向边也遵从这个规则就行。)

#include
#pragma GCC optimize("Ofast")
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
//#define int long long
#define pii pair
#define pdd pair
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=2e5+5;
const int inf=0x3f3f3f3f;
const int mod=998244353;
const double eps=1e-6;
const double PI=acos(-1);
vector<int>e[N];int in[N];int p[N];
signed main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,m;
        cin>>n>>m;
        vector<pii>ans;
        for(int i=1;i<=n;i++)
        {
            e[i].clear();
            in[i]=0;
            p[i]=0;
        }
        for(int i=1;i<=m;i++)
        {
            int op,u,v;
            cin>>op>>u>>v;
            ans.pb(mp(u,v));
            if(op==1)
            {
                e[u].pb(v);
                in[v]++;
            }
        }
        queue<int>q;
        for(int i=1;i<=n;i++)
        {
            if(in[i]==0)
                q.push(i);
        }
        int tot=0;
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            p[u]=++tot;
            for(auto i:e[u])
            {
                in[i]--;
                if(in[i]==0)
                    q.push(i);
            }
        }
        if(tot!=n)
            cout<<"NO"<<endl;
        else
        {
            cout<<"YES"<<endl;
            for(auto i:ans)
            {
                int u=i.fi,v=i.se;
                if(p[u]<p[v])
                    cout<<u<<' '<<v<<endl;
                else
                    cout<<v<<' '<<u<<endl;
            }
        }
    }
}

你可能感兴趣的:(拓扑排序)