码迷,mamicode.com
首页 > 其他好文 > 详细

【poj】 Roads in the North 题解

时间:2018-10-12 17:03:53      阅读:130      评论:0      收藏:0      [点我收藏+]

标签:main   树的直径   problem   端点   push   ++   code   cstring   empty   

题目链接:http://poj.org/problem?id=2631

求树的直径模板。

定理:

树上任意一个点的在树上的最长路一定以树的直径的两端点其中一点结束。

做法:

两边bfs,第一次先找到node(树的直径的两端点其中一个),再一次求node的最长路所结束的点t
node—>t就是树的直径

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int n, dis[maxn], ans, node;
bool vis[maxn];
struct edge{
    int from, to, next, len;
}e[maxn<<2];
int cnt, head[maxn];
void add(int u, int v, int w)
{
    e[++cnt].from = u;
    e[cnt].next = head[u];
    e[cnt].len = w;
    e[cnt].to = v;
    head[u] = cnt;
}
queue<int> q;
void bfs(int s)
{
    vis[s] = 1;
    q.push(s);
    while(!q.empty())
    {
        int now = q.front(); q.pop();
        for(int i = head[now]; i != -1; i = e[i].next)
        {
            if(vis[e[i].to] == 0)
            {
                dis[e[i].to] = dis[now] + e[i].len;
                vis[e[i].to] = 1;
                q.push(e[i].to);
                if(dis[e[i].to] > ans)
                {
                    ans = dis[e[i].to];
                    node = e[i].to;
                }
            }
        }
    }
}
int main()
{
    memset(head, -1, sizeof(head));
    int u, v, w;
    while(scanf("%d%d%d",&u, &v, &w) == 3)
    {
        add(u, v, w);
        add(v, u, w);
    }
    memset(dis, 0, sizeof(dis));
    memset(vis, 0, sizeof(vis));
    ans = 0;
    bfs(1);
    
    memset(vis, 0, sizeof(vis));
    dis[node] = 0;
    ans = 0;
    bfs(node);
    
    printf("%d\n", ans);
    return 0;
}

【poj】 Roads in the North 题解

标签:main   树的直径   problem   端点   push   ++   code   cstring   empty   

原文地址:https://www.cnblogs.com/MisakaAzusa/p/9778945.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!