给定一幅无向带权连通图G = (V, E) (这里V是点集,E是边集)。从点u开始的最短路径树是这样一幅图G1 = (V, E1),其中E1是E的子集,并且在G1中,u到所有其它点的最短路径与他在G中是一样的。
现在给定一幅无向带权连通图G和一个点u。你的任务是找出从u开始的最短路径树,并且这个树中所有边的权值之和要最小。
收起
输入
单组测试数据。
第一行有两个整数n和m(1 ≤ n ≤ 310^5, 0 ≤ m ≤ 310^5),表示点和边的数目。
接下来m行,每行包含3个整数 ui, vi, wi ,表示ui和vi之间有一条权值为wi的无向边(1 ≤ ui,vi ≤ n, 1 ≤ wi ≤ 10^9)。
输入保证图是连通的。
最后一行给出一个整数u (1 ≤ u ≤ n),表示起点。
输出
输出这棵树的最小的权值之和。
输入样例
3 3
1 2 1
2 3 1
1 3 2
3
输出样例
2
题意:
找出从u开始的最短路径树,并且这个树中所有边的权值之和要最小。
思路:
找最短路径并记录每个点的的最小前驱边,最后把它们加起来并输出。
AC代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#include<cstdio>
using namespace std;
typedef long long ll;
const ll N = 3e5+5;
const ll M = 1e6;
ll d[N];
bool vis[N];
ll head[N],tot;
ll p[N];
ll v[N];
ll f[N];
struct Edge
{
ll next;
ll to;
ll dis;
}edge[N*2];
inline void add(ll from,ll to,ll dis)
{
edge[++tot].next = head[from];
edge[tot].to = to;
edge[tot].dis = dis;
head[from] = tot;
}
priority_queue< pair<ll, int> >q;
void dij(int k)
{
memset(d,0x3f3f3f,sizeof(d));
memset(v,0,sizeof v);
d[k]=0;
q.push(make_pair(0,k));
while(q.size())
{
ll x=q.top().second;
q.pop();
//if(d[x]<q.top().first)continue;
if(v[x])continue;
v[x]=1;
for(int i=head[x];i;i=edge[i].next)
{
ll y=edge[i].to,z=edge[i].dis;
if(d[y]>d[x]+z)
{
f[y]=z;
d[y]=d[x]+z;
q.push(make_pair(-d[y],y));
}
else
if(d[y]==d[x]+z)
{
f[y]=min(f[y],z);
}
}
}
}
int main()
{
int n,m;
scanf("%d%d", &n, &m);
for(int i=0;i<m;i++)
{
ll x,y,z;
scanf("%lld%lld%lld", &x, &y, &z);
add(x,y,z);
add(y,x,z);
}
int k;
scanf("%d",&k);
dij(k);
ll ans=0;
for(int i=1;i<=n;i++)
ans+=f[i];
printf("%lld\n", ans);
}