sdut3045迷之圖論--(多叉樹求最長鏈)

標籤：style   blog   io   color   sp   for   on   資料   div   

迷之圖論Time Limit: 1000MS Memory limit: 65536K題目描述

FF是圖論高手，所以我要出圖論且不出流問題。

給出一個樹，求樹的最長鏈的長度。

輸入 多組輸入。每組輸入的第一行為n(1 <= n <= 100000),代表節點個數，節點編號從1 到n，接下來的n-1行，每行兩個正整數u，v,代表u，v之間有一條邊相連。保證每組資料都是一棵樹。

輸出 對於每組資料，輸出一個正整數代表答案。

樣本輸入

121 2

樣本輸出

12

提示 來源 zmxdfs搜尋

#include <cstdio>#include <cstring>#include <algorithm>using namespace std ;struct node{    int v ;    int next ;}p[1000000];int head[110000] , cnt , vis[110000] ;int ans , max1[110000] ;void add(int u,int v){    p[cnt].v = v ;    p[cnt].next = head[u] ; head[u] = cnt++ ;    p[cnt].v = u ;    p[cnt].next = head[v] ; head[v] = cnt++ ;}void dfs(int u,int num){    int i , j , v , flag = 0 , k1 = -1 , k2 = -1 ;    for(i = head[u] ; i != -1 ; i = p[i].next)    {        v = p[i].v ;        if( !vis[v] )        {            flag = 1 ;            vis[v] = num+1 ;            dfs(v,num+1) ;        }    }    if( flag == 0 )    {        max1[u] = 1 ;        return ;    }    for(i = head[u] ; i != -1 ; i = p[i].next)    {        v = p[i].v ;        if( vis[u]+1 != vis[v] )            continue ;        if( k1 == -1 )            k1 = max1[v] ;        else        {            if( max1[v] > k1 )            {                k2 = k1 ;                k1 = max1[v] ;            }            else if(  max1[v] > k2 )                k2 = max1[v] ;        }    }    max1[u] = k1 + 1 ;    ans = max( ans,max1[u] ) ;    if( k2 == -1 )        return ;    else    {        ans = max( ans,k1+k2+1 ) ;    }    return ;}int main(){    int n , u , v , i ;    while( scanf("%d", &n) != EOF )    {        cnt = 0 ;        ans = 2 ;        memset(head,-1,sizeof(head)) ;        memset(vis,0,sizeof(vis)) ;        if(n == 1)        {            printf("1\n") ;            continue ;        }        for(i = 1 ; i < n ; i++)        {            scanf("%d %d", &u, &v) ;            add(u,v) ;        }        vis[1] = 1 ;        dfs(1,1);        printf("%d\n", ans) ;    }    return 0;}

 

 

sdut3045迷之圖論--(多叉樹求最長鏈)