树-图的直径

定义

所有点 距离最远的点 是直径的两个端点之一
两次求解最远可以求解到A,B 当然答案不唯一
一次求解的 反例
A
| |
D
||
K
| |
// \
B C
对于 A来说 B和C都是最远 如果 B C为直径的话 A-D-K-B明显大于B-K-C
则再次考虑B-A

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
const int N = 200005;
int f[N];
int lvl[N];
int flag[N];
int cnt=0;
int n;
int f2[N];
vector<pii> graph[N];
int bfs(int beg)
{
    int maxn = 0;

    int index = -1;
    memset(f,0,sizeof(f));
    memset(lvl,0,sizeof(lvl));
    memset(flag,0,sizeof(flag));
    queue<int> q;
    q.push(beg);
    while(!q.empty()){
        int u = q.front();
        q.pop();
        if(flag[u])
            continue;
        //cout<<u<<endl;
        flag[u] = 1;
        for(auto it = graph[u].begin(); it != graph[u].end(); ++it){
            int v = it->first;
            int w = it->second;
            if(flag[v])
                continue;
            lvl[v] = lvl[u] +1;
            f[v] = f[u] + w;
            if(f[v] > maxn){
                maxn = f[v];
                cnt = lvl[v];
                index = v;
            }
            q.push(v);
        }
    }
   // for(int i=1; i<=n; ++i)
     //   cout<<f[i]<<endl;
    return index;

}

int main()
{
    ios::sync_with_stdio(false);

    cin>>n;
    for(int i=0; i< n - 1; ++i){
        int u,v,w;
        cin>>u>>v>>w;
        graph[u].push_back(make_pair(v,w));
        graph[v].push_back(make_pair(u,w));
    }
    int k = bfs(1);
    //cout<<endl;
    //cout<<a<<endl;
    int v = bfs(k);
    // from k to v & v to k
    //cout<<v<<" "<<a<<endl;
    for(int i=1; i<=n; ++i)
    {
        f2[i] = f[i];
        //cout<<f[i]<<" ";
    }
  //  cout<<endl;
    int u = bfs(v);
   // cout<<v<<endl;

    int sum = 0;
    for(int i=1; i<=n; ++i){
        if( i == v)
            continue;
        sum += max(f2[i],f[i]);
        //cout<<sum<<endl;
    }
    cout<<sum<<endl;
    //cout<<cnt<<endl;
    //sum -= f[v];
    //cout<<sum<<endl;


}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 专业考题类型管理运行工作负责人一般作业考题内容选项A选项B选项C选项D选项E选项F正确答案 变电单选GYSZ本规程...
    小白兔去钓鱼阅读 9,059评论 0 13
  • 在C语言中,五种基本数据类型存储空间长度的排列顺序是: A)char B)char=int<=float C)ch...
    夏天再来阅读 3,418评论 0 2
  • 高级钳工应知鉴定题库(858题) ***单选题*** 1. 000003难易程度:较难知识范围:相关4 01答案:...
    开源时代阅读 5,973评论 1 9
  • 选择题部分 1.(),只有在发生短路事故时或者在负荷电流较大时,变流器中才会有足够的二次电流作为继电保护跳闸之用。...
    skystarwuwei阅读 13,447评论 0 7
  • 一日,吃早餐时,同事马老师突然嚷嚷了一句:胡瑾慎在海口那边跳了海!大家都有点蒙,怎么回事!突然,又象想起了什么...
    听雨江南久阅读 1,202评论 15 36