最短路问题
属于修改最短路更新操作的题目
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define maxn 10010
struct node
{
int v;
int w;
node(int _v=0,int _w=0):v(_v),w(_w) {};
bool operator < (const node &a) const
{
return w>a.w;
}
};
struct Edge
{
int v;
int w;
Edge(int _v=0,int _w=0):v(_v),w(_w) {};
};
vector<Edge> E[maxn];
int dis[maxn];
int prec[maxn];
bool vis[maxn];
int n,m;
void addedge(int u,int v,int w)
{
E[u].push_back(Edge(v,w));
E[v].push_back(Edge(u,w));
}
void djs(int st)
{
memset(dis,inf,sizeof dis);
memset(vis,false,sizeof vis);
memset(prec,inf,sizeof prec);
priority_queue<node> q;
node a(st,0);
q.push(a);
node t;
int cnt=0;
while(!q.empty())
{
t=q.top();
q.pop();
int u=t.v;
int w=t.w;
if(vis[u]) continue;
vis[u]=true;
dis[u]=w;
cnt++;
for(int i=0;i<E[u].size();i++)
{
if(!vis[E[u][i].v])
{
if(dis[E[u][i].v]>dis[u]+E[u][i].w)
{
dis[E[u][i].v]=dis[u]+E[u][i].w;
prec[E[u][i].v]=E[u][i].w;
q.push(node(E[u][i].v,dis[E[u][i].v]));
}
else if(dis[E[u][i].v]==dis[u]+E[u][i].w)
{
prec[E[u][i].v]=min(prec[E[u][i].v],E[u][i].w);
}
}
}
if(cnt==n) break;
}
}
int main()
{
cin>>n>>m;
for(int i=1;i<=m;i++)
{
int u,v,w;
cin>>u>>v>>w;
addedge(u,v,w);
}
djs(1);
int ans=0;
for(int i=2;i<=n;i++)
ans+=prec[i];
cout<<ans<<endl;
return 0;
}