编程语言
首页 > 编程语言> > 《算法竞赛进阶指南》0x21树和图的遍历 求dfs序以及树的重心

《算法竞赛进阶指南》0x21树和图的遍历 求dfs序以及树的重心

作者:互联网

#include<iostream>
#include<string.h>
using namespace std;
#define maxn 100
int ver[maxn],head[maxn],nxt[maxn],size[maxn],len[maxn];
int n,m;
int vis[maxn];
int tot=0;
void addedge(int u,int v,int w){
    ver[++tot]=v;
    len[tot]=w;
    nxt[tot]=head[u];
    head[u]=tot;
}
int ans=0x7fffffff,pos;
void dfs(int x){
    vis[x]=1;
    size[x]=1;
    int max_part=0;
    for(int i=head[x];i;i=nxt[i]){
        int v=ver[i];
        if(vis[v])continue;
        vis[v]=1;
        dfs(v);
        size[x]+=size[v];
        max_part=max(max_part,size[v]);
    }
    max_part=max(max_part,n-size[x]);
    if(max_part<ans){
        ans=max_part;
        pos=x;
    }
}
void init(){
    cin>>n>>m;
    tot=0;
    memset(vis,0,sizeof(vis));
    int u,v,w;
    for(int i=0;i<m;i++){
        cin>>u>>v>>w;
        addedge(u,v,w);
        addedge(v,u,w);
    }
    dfs(1);
    cout<<"树的重心:"<<pos<<endl;
    cout<<"划分的最大子树大小:"<<ans<<endl;
}
int d[maxn];
int dfn=0;
void dfs2(int x){//获取dfs序 
    d[dfn++]=x;//记录第一次访问的时间戳 
    vis[x]=1;
    for(int i=head[x];i;i=nxt[i]){
        int y=ver[i];
        if(vis[y])continue;
        dfs2(y);
    }
    d[dfn++]=x;//回溯时刻记录访问的编号 
}
int main(){
    init();
    memset(vis,0,sizeof(vis));
    dfs2(1);
    for(int i=0;i<dfn;i++)cout<<d[i]<<" ";
}
9 8
1 2 1
1 7 1
1 4 1
2 8 1
2 5 1
4 3 1
4 6 1
3 9 1

 

标签:树和图,进阶,int,max,0x21,tot,vis,maxn,size
来源: https://www.cnblogs.com/randy-lo/p/13158499.html