Codeforces Round #303 (Div. 2) E. Paths and Trees（贪心+最短路）

题目链接


思路：在保证是最短路的同时如果有多条最短路就选边权小的那个。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn =3e5+1;
const  ll inf=1e18;
struct node{
	int u,w,id;
};
vector<node>g[maxn];
int ans[maxn];
ll sum=0,d[maxn],val[maxn];
bool vis[maxn];
void spfa(int s)
{
	memset(vis,false,sizeof(vis));
	fill(d,d+maxn,inf);
	d[s]=0;
	queue<int>q;
	q.push(s);
	vis[s]=true;
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		for(auto v:g[top])
		{
			if(d[top]+v.w<d[v.u])
			{
				d[v.u]=d[top]+v.w;
				ans[v.u]=v.id;
				q.push(v.u);
			}
			else if(d[top]+v.w==d[v.u])
			{
				if(v.w<val[ans[v.u]]) 
				{
				ans[v.u]=v.id;
				q.push(v.u);
				}
			}
		}
	}
}
int main()
{
	int n,m,u,v,w,s;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;++i)
	{
		scanf("%d %d %d",&u,&v,&w);
		g[u].push_back({v,w,i});
		g[v].push_back({u,w,i});
		val[i]=w;
	}
	scanf("%d",&s);
	spfa(s);
	for(int i=1;i<=n;++i) if(ans[i]) sum+=val[ans[i]];
	printf("%lld\n",sum);
	for(int i=1;i<=n;++i)
	if(ans[i]) printf("%d ",ans[i]);
}