牛客练习赛27——C——最短路变形

题目链接：牛客练习赛27

题目描述：
小w有一张n个点n-1条边的无向联通图,每个点编号为1~n,每条边都有一个长度
小w现在在点x上
她想知道从点x出发经过每个点至少一次，最少需要走多少路。

题目思路：比赛的时候确实欠缺考虑，只发现有些边会经过两次，有些边只用经过一次，只要经过一次的边权值最大就行。再仔细观察就会发现，从任意点出发，遍历所有点，都会只有一条路只经过一次，可以画图模拟一下，因为这个题的图应该是一个树形结构。所以我们要只经过一次的路径是最长路，再用权值之和减去最长路长度就行。

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

const int maxn=50000+50;

typedef long long ll;

const ll INF=0x3f3f3f3f;

struct Edge
{
    int from;
    int to;
    int next;
    ll w;
    Edge(int u=0,int v=0,ll cost=0):from(u),to(v),w(cost){}
}edges[maxn*2];

struct HeapNode
{
    ll dist;
    int u;
    bool operator <(const HeapNode &rhs)const
    {
        return dist>rhs.dist;
    }
    HeapNode(int _u=0,ll d=0):u(_u),dist(d){}
};

int head[maxn],tot;

bool vis[maxn];

ll dis[maxn];//最短距离

void init()
{
    memset(head,-1,sizeof(head));
    tot=0;
}

void add_edges(int u,int v,int cost)
{
    edges[tot].from=u;
    edges[tot].to=v;
    edges[tot].w=cost;
    edges[tot].next=head[u];
    head[u]=tot++;
}

void dfs(int s,int n)
{
    priority_queue<HeapNode>Q;
    for(int i=0;i<=n;i++)dis[i]=INF;
    dis[s]=0;
    memset(vis,false,sizeof(vis));
    Q.push(HeapNode(s,0));
    while(!Q.empty())
    {
        HeapNode x=Q.top();Q.pop();
        int u=x.u;
        if(vis[u])continue;
        vis[u]=true;
        for(int i=head[u];i!=-1;i=edges[i].next)
        {
            int v=edges[i].to;
            if(dis[v]>dis[u]+edges[i].w)
            {
                //cout<<"OK"<<endl;
                dis[v]=dis[u]+edges[i].w;
                Q.push(HeapNode(v,dis[v]));
            }
        }
    }
}

int main()
{
    int n,t;
    while(scanf("%d%d",&n,&t)!=EOF)
    {
        init();
        int u,v;
        ll w;
        ll ans0=0;
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d%lld",&u,&v,&w);
            ans0+=w*2;
            add_edges(u,v,w);
            add_edges(v,u,w);
        }
        //cout<<ans0<<endl;
        dfs(t,n);
        ll ans=0;
        for(int i=1;i<=n;i++)
        {
            //printf("%lld   ",dis[i]);
            //printf("\n");
            ans=max(ans,dis[i]);
        }
        printf("%lld\n",ans0-ans);
    }
    return 0;
}