POJ2631 Roads in the North

本文介绍了一种使用两次广度优先搜索(BFS)来求解树的直径的方法。首先从任意节点开始进行一次BFS找到距离最远的节点,再以此节点为起点进行第二次BFS,最终得到的距离即为树的直径。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

树的直径,两遍bfs实现。


#include <cstdio>
#include <cstring>
const int MAXN = 10010;

struct Edge
{
    int v, w, next;
}edge[MAXN * 2];
int edgeNumber, head[MAXN];
int destination, distance;

inline void addEdge(int u, int v, int w)
{
    edge[edgeNumber].v = v;
    edge[edgeNumber].w = w;
    edge[edgeNumber].next = head[u];
    head[u] = edgeNumber ++;
}

int dist[MAXN];
int queue[MAXN];
int front, rear;
bool visit[MAXN];

void bfs(int start)
{
    memset(visit, false, sizeof(visit));
    memset(dist, -1, sizeof(dist));
    queue[0] = start;
    visit[start] = true;
    dist[start] = 0;
    front = 0, rear = 1;
    while(front < rear)
    {
        int u = queue[front++];
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v = edge[i].v;
            int w = edge[i].w;
            if(dist[v] == -1 || dist[v] > dist[u] + w)
            {
                dist[v] = dist[u] + w;
                if(!visit[v])
                {
                    visit[v] = true;
                    queue[rear++] = v;
                }
            }
        }
    }
    destination = 1;
    distance = 0;
    for(int i=0;i<MAXN;++i)
    {
        if(dist[i] > distance)
        {
            distance = dist[i];
            destination = i;
        }
    }
}

int main()
{
    int u, v, w;
    edgeNumber = 0;
    memset(head, -1, sizeof(head));
    while(~scanf("%d%d%d",&u,&v,&w))
    {
        addEdge(u, v, w);
        addEdge(v, u, w);
    }
    bfs(1);
    bfs(destination);
    printf("%d\n", distance);
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值