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

【Spfa/最短路模板】遍历所有点的最短路径

时间:2019-01-22 20:13:34      阅读:232      评论:0      收藏:0      [点我收藏+]

标签:define   最短路   sizeof   add   题目   一个   inf   旅游   str   

题目描述

明明暑假来济南旅游旅游,他打算游玩N个旅游景点,N-1条双向连接的道路将它们联通起来,每一条道路有固定长度。一开始明明位于1号景点。
现在希望你能够求出旅行长度最小的方案,使得每个景点至少被访问到一次。
 

 

输入

第一行两个整数N,代表景点数目。
接下来N-1行,每行三个整数s, t, w,表示有一条从s到t的双向道路,长度为w。s和t的编号从1开始。

 

输出

一行一个整数,代表能够访问每个景点至少一次的方案的最小旅行长度。
 
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=50005;
struct Edge
{
    int v;
    int cost;
    Edge(int _v=0,int _cost=0):v(_v),cost(_cost){}
};
vector<Edge> E[maxn];
void addedge(int u,int v,int w)
{
    E[u].push_back(Edge(v,w));
}
bool vis[maxn];
int cnt[maxn];
int dist[maxn];
bool spfa(int st,int n)
{
    memset(vis,false,sizeof vis);
    for(int i=1;i<=n;i++)
        dist[i]=INF;
    vis[st]=true;
    dist[st]=0;
    queue<int>qu;
    while(!qu.empty())
    {
        qu.pop();
    }
    qu.push(st);
    memset(cnt,0,sizeof cnt);
    cnt[st]=1;
    while(!qu.empty())
    {
        int u=qu.front();
        qu.pop();
        vis[u]=false;
        for(int i=0;i<E[u].size();i++)
        {
            int v=E[u][i].v;
            if(dist[v]>dist[u]+E[u][i].cost)
            {
                dist[v]=dist[u]+E[u][i].cost;
                if(!vis[v])
                {
                    vis[v]=true;
                    qu.push(v);
                    if(++cnt[v]>n)
                        return false;
                }
            }
        }
    }
    return true;
}
int vis0[50005];
int main()
{
    int n;
    scanf("%d",&n);
    int u, v, w;
    int ant=0,mx=0;
    for (int i = 1; i <= n-1; i++)
    {
        scanf("%d %d %d",&u,&v,&w);
        ant+=w;
        addedge(u,v,w);
        addedge(v,u,w);
        vis0[u]=0;
        vis0[v]=1;
    }
    ant*=2;
    spfa(1,n);
    for(int i=1;i<=n;i++)
    {
        mx=max(mx,dist[i]);
    }
    printf("%d\n",ant-mx);
    return 0;
}

 

【Spfa/最短路模板】遍历所有点的最短路径

标签:define   最短路   sizeof   add   题目   一个   inf   旅游   str   

原文地址:https://www.cnblogs.com/Diliiiii/p/10305755.html

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