tarjin例题——上白慧泽音

题目描述

在幻想乡，上白泽慧音是以知识渊博闻名的老师。春雪异变导致人间之里的很多道路都被大雪堵塞，使有的学生不能顺利地到达慧音所在的村庄。因此慧音决定换一个能够聚集最多人数的村庄作为新的教学地点。人间之里由N个村庄（编号为1..N）和M条道路组成，道路分为两种一种为单向通行的，一种为双向通行的，分别用1和2来标记。如果存在由村庄A到达村庄B的通路，那么我们认为可以从村庄A到达村庄B，记为(A,B)。当(A,B)和(B,A)同时满足时，我们认为A,B是绝对连通的，记为<A,B>。绝对连通区域是指一个村庄的集合，在这个集合中任意两个村庄X,Y都满足<X,Y>。现在你的任务是，找出最大的绝对连通区域，并将这个绝对连通区域的村庄按编号依次输出。若存在两个最大的，输出字典序最小的，比如当存在1,3,4和2,5,6这两个最大连通区域时，输出的是1,3,4。

输入输出格式

输入格式：

 

第1行：两个正整数N,M

第2..M+1行：每行三个正整数a,b,t, t = 1表示存在从村庄a到b的单向道路，t = 2表示村庄a,b之间存在双向通行的道路。保证每条道路只出现一次。

 

输出格式：

 

第1行： 1个整数，表示最大的绝对连通区域包含的村庄个数。

第2行：若干个整数，依次输出最大的绝对连通区域所包含的村庄编号。

 

输入输出样例

输入样例#1： 复制

5 5
1 2 1
1 3 2
2 4 2
5 1 2
3 5 1

输出样例#1： 复制

3
1 3 5

说明

对于60%的数据：N <= 200且M <= 10,000

对于100%的数据：N <= 5,000且M <= 50,000

 

 

代码：

#include<bits/stdc++.h>
using namespace std;
int n,m,v[5005],a,b,c,in,dfn[5005],low[5005];
vector<int> G[5005];
stack<int> s;
vector<set<int> > res;
bool cmp(set<int>a,set<int>b){if(a.size()==b.size())return *(a.begin())<*(b.begin());return a.size()>b.size();}
void tarjin(int x) {
    s.push(x);
    in++;
    v[x]=1;
    low[x]=in;
    dfn[x]=in;
    for(int i=0;i<G[x].size();i++){
        if(!dfn[G[x][i]]){
            tarjin(G[x][i]);
            low[x]=min(low[x],low[G[x][i]]);
        }else if(v[G[x][i]]){
            low[x]=min(low[x],dfn[G[x][i]]);
        }
    }
    if(low[x]==dfn[x]){
        v[x]=0;
        set<int> ans;
        while(s.top()!=x){
            ans.insert(s.top());
            v[s.top()]=0;
            s.pop();
        }
        ans.insert(x);
        s.pop();
        res.push_back(ans);
    }
}
int  main(){
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>a>>b>>c;
        if(c==2) G[b].push_back(a);
        G[a].push_back(b);
    }
    for(int i=1;i<=n;i++){
        if(!dfn[i]){
            tarjin(i);
        }
    }
    sort(res.begin(),res.end(),cmp);
    set<int> ans=res[0];
    cout<<ans.size()<<endl;
    for(set<int>::iterator it=ans.begin();it!=ans.end();it++){
        cout<<*it<<" ";
    }
    return 0;
}