NC99 树的直径

NC99 树的直径

这道题有个坑。

这道题不是二叉树，而且还需要你自己建图。而且不可以使用二维数组建图，因为n太大。要使用领接表。

同时我第一次想的是最长的路径必定是在一个叶子节点到另一个叶子节点，我找遍所有的取最大的，然后去最大的就行了，那么有几个子叶节点就需要几个dfs，那么这样还是会超时。

于是，就有了，先随便找一个点搜到他的最远点，然后在从最原来搜，这次搜到就是最大的了。

下面是我两次的代码，第一个是超时的。

/**
 * struct Interval {
 *	int start;
 *	int end;
 * };
 */

class Solution {
public:
    
    
    int maxn=0;
    //int a[2000][2000];
    vector< pair<int,int> > a[20000];
    void dfs( int cur,int pre,int sum)
    {
        
        int flag=0;
        if(a[cur].size()==1)
        {
            maxn=max(maxn,sum);
            return ;
        }
        
        else{
            for(int i=0;i<a[cur].size();i++)
            {
                
                int j=a[cur][i].first;
                if(j!=pre)
                dfs(j,cur,sum+a[cur][i].second);
            }
        }
      
        

        
        
    }
    int solve(int n, vector<Interval>& Tree_edge, vector<int>& Edge_value) {
        // write code here
        
        for(int i=0;i<n-1;i++)
        {
            int x=Tree_edge[i].start;
            int y=Tree_edge[i].end;
            int z=Edge_value[i];
            
            a[x].push_back(make_pair(y, z));
            a[y].push_back(make_pair(x, z));
         
            
        }
        
        for(int i=0;i<n;i++)
        {
            if(a[i].size()==1)
            {
                dfs(a[i][0].first,i,0+a[i][0].second);
            }
        }
        
        return maxn;
        
    }
};

class Solution {
public:
    
    
    int maxn=0,u=-1;
    //int a[2000][2000];
    vector< pair<int,int> > a[20000];
    void dfs( int cur,int pre,int sum)
    {
        
        int flag=0;
        if(a[cur].size()==1&&a[cur][0].first==pre)
        {
            if(maxn<sum)
            {
                maxn=sum;
                u=cur;
            }
            return ;
        }
        
        else{
            for(int i=0;i<a[cur].size();i++)
            {
                int j=a[cur][i].first;
                if(j!=pre)
                dfs(j,cur,sum+a[cur][i].second);
            }
        }
      
        

        
        
    }
    int solve(int n, vector<Interval>& Tree_edge, vector<int>& Edge_value) {
        // write code here
        
        for(int i=0;i<n-1;i++)
        {
            int x=Tree_edge[i].start;
            int y=Tree_edge[i].end;
            int z=Edge_value[i];
            
            a[x].push_back(make_pair(y, z));
            a[y].push_back(make_pair(x, z));
         
            
        }
        

        dfs(0,-1,0);
        dfs(u,-1,0);
        
        return maxn;
        
    }
};