1021 Deepest Root (25 分)连通图个数、图的遍历dfs

①测试点3超时，我没有考虑有环的情况，拓扑排序好像是可以判断有向无环图吧，但题目给出了一定是n个顶点，n-1条边，如果有环，则一定不连通。应该是for dfs用太多啦
②memset的头文件是string.h，fill是algorithm
③并查集可以找两个集合的共同元素，也可以合并，但是求连通图个数，如果需要dfs，用这个最简单

 for(int i=1;i<=n;i++){
        if(vis[i]==false){
            dfs(i,1);
            block++;
        }
    }

#include<iostream>
#include<set>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=10001;
vector<int> vi[N];
set<int> s;
int maxheight=0;
bool vis[N]={false};
void dfs(int v,int h){
    if(h>maxheight){
        s.clear();
        s.insert(v);
        maxheight=h;
    }
    else if(h==maxheight){
        s.insert(v);
    }
    vis[v]=true;
    for(int i=0;i<vi[v].size();i++){
        if(vis[vi[v][i]]==false)
            dfs(vi[v][i],h+1);
    }
}

int main(){
    int n,block=0;
    cin>>n;
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        vi[a].push_back(b);
        vi[b].push_back(a);
    }
    for(int i=1;i<=n;i++){
        if(vis[i]==false){
            dfs(i,1);
            block++;
        }
    }
    if(block>1) cout<<"Error: "<<block<<" components"<<endl;
    else
    {
        for(int i=1;i<=n;i++){
            if(vis[i]==false){
                dfs(i,1);
                block++;
            }
            memset(vis,false,sizeof(vis));
        }
        for(auto it=s.begin();it!=s.end();it++)
            cout<<*it<<endl;
    }
    return 0;
}

然后修改后，结果是不输出5

#include<iostream>
#include<set>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=10001;
vector<int> vi[N];
set<int> s;
int maxheight=0;
bool vis[N]={false};
void dfs(int v,int h){
    if(h>maxheight){
        s.clear();
        s.insert(v);
        maxheight=h;
    }
    else if(h==maxheight){
        s.insert(v);
    }
    vis[v]=true;
    for(int i=0;i<vi[v].size();i++){
        if(vis[vi[v][i]]==false)
            dfs(vi[v][i],h+1);
    }
}

int main(){
    int n,block=0;
    cin>>n;
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        vi[a].push_back(b);
        vi[b].push_back(a);
    }
    for(int i=1;i<=n;i++){
        if(vis[i]==false){
            dfs(i,1);
            block++;
        }
    }
    if(block>1) cout<<"Error: "<<block<<" components"<<endl;
    else
    {
        set<int>::iterator it=s.begin();
        memset(vis,false,sizeof(vis));
        dfs(*it,1);
        for(it=s.begin();it!=s.end();it++)
            cout<<*it<<endl;
    }
    return 0;
}

原因：第一次dfs找到的是离1最远的点，得到了maxheight，但这只能得到该点一定是deepest，并且在该点的另一头也是deepest，从该点出发，得到的树高可能会大于第一次的maxheight，所以会清空第一次的结果。应该取并集

#include<iostream>
#include<set>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=10001;
vector<int> vi[N];
set<int> s,s1;
int maxheight=0;
bool vis[N]={false};
void dfs(int v,int h){
    if(h>maxheight){
        s.clear();
        s.insert(v);
        maxheight=h;
    }
    else if(h==maxheight){
        s.insert(v);
    }
    vis[v]=true;
    for(int i=0;i<vi[v].size();i++){
        if(vis[vi[v][i]]==false)
            dfs(vi[v][i],h+1);
    }
}

int main(){
    int n,block=0;
    cin>>n;
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        vi[a].push_back(b);
        vi[b].push_back(a);
    }
    for(int i=1;i<=n;i++){
        if(vis[i]==false){
            dfs(i,1);
            block++;
        }
    }
    if(block>1) cout<<"Error: "<<block<<" components"<<endl;
    else
    {
        set<int>::iterator it;
        for( it=s.begin();it!=s.end();it++)
            s1.insert(*it);
        it=s.begin();
        memset(vis,false,sizeof(vis));
        dfs(*it,1);
        for( it=s.begin();it!=s.end();it++)
            s1.insert(*it);
        for(auto it1=s1.begin();it1!=s1.end();it1++)
            cout<<*it1<<endl;
    }
    return 0;
}

不能清空vis对每一个v都dfs找到最大，虽然答案没问题，但是会超时。