2020 蓝桥杯省赛 B 组模拟赛（一） 程序设计：最短路

•  1000ms
•  262144K

给定一个 nn 点 mm 边的有向带权图表示一座城市，起点为 11 。送餐小哥需要给 nn 个客户送外卖，第 ii 个客户的家在第 ii 号点。由于他的车子容量很小，所以一次只能容纳一份外卖，所以送达外卖之后就要回到起点取新的外卖送下一单，直到全部送到位置。

有向图保证联通。外卖小哥一定走的最短路。

求送餐小哥走的总路程。

输入格式

第一行一个整数 TT，表示数据组数。

对于每组数据，第一行两个整数 nn 和 mm 。

接下来 mm 行，每行三个整数 u_i,v_i,c_iui​,vi​,ci​ 表示每条有向边。

输出格式

对于每组数据，输出一行一个整数表示答案。

数据范围

对于 20\%20% 的数据： 0 < n \leq 1000<n≤100 ,

对于 40\%40% 的数据： 0 < n \leq 3000<n≤300 ,

对于 60\%60% 的数据： 0 < n \leq 10000<n≤1000 ,

对于 100\%100% 的数据： 0 < n \leq 20000, m \leq 60000, 1\le T\le 10,0\le c_i\le 10^9,1\le u_i,v_i\le n0<n≤20000,m≤60000,1≤T≤10,0≤ci​≤109,1≤ui​,vi​≤n

保证答案在 `long long` 范围内。

样例输入复制

2
2 2
1 2 13
2 1 33
4 6
1 2 10
2 1 60
1 3 20
3 4 10
2 4 5
4 1 50

样例输出复制

46
210

思路：两次建图，两次迪杰斯特拉求最短路即可。

注意：n,m范围有问题，按题目中的范围会段错误。

#include<iostream>
#include<vector>
#include<queue>
#include<unordered_map>
#include<cstring>

using namespace std;

typedef long long ll;
typedef pair<ll, int> P;
const int N = 20010, M = 60010;

int idx, h[N], e[M], w[M], ne[M];
ll dis[N];
int vis[N];

struct node{
	int x,y,z;
}nod[N];

void add(int a, int b, int c)
{
	e[idx] = b;
	w[idx] = c;
	ne[idx] = h[a];
	h[a] = idx++;
}

void dij()
{
	memset(dis, 0x3f, sizeof dis);
	memset(vis, 0, sizeof vis);
	dis[1] = 0;
	
	priority_queue<P, vector<P>, greater<P>> q;
	q.push({0,1});
	
	while(!q.empty())
	{
		P p = q.top();
		q.pop();
		
		int u = p.second;
		ll l = p.first;
		
		if(vis[u]) continue;
		vis[u] = 1;
		
		for(int i=h[u]; i!=-1; i=ne[i])
		{
			int v = e[i];
			if(dis[v] > dis[u]+w[i])
			{
				dis[v] = dis[u]+w[i];
				q.push({dis[v], v});
			}
		}
	}
	
	
}

int main()
{
	int t;
	cin>>t;
	while(t--)
	{	
		int n,m;
		cin>>n>>m;
		
		idx = 0;
		memset(h, -1, sizeof h);
		for(int i=1; i<=m; i++)
		{
			cin>>nod[i].x>>nod[i].y>>nod[i].z;
			add(nod[i].x,nod[i].y,nod[i].z);
		}
		dij();
		
		ll ans = 0;
		for(int i=2; i<=n; i++) ans += dis[i];
		
		idx = 0;
		memset(h, -1, sizeof h);
		for(int i=1; i<=m; i++)
		{
			add(nod[i].y,nod[i].x,nod[i].z);
		}
		dij();
		
		for(int i=2; i<=n; i++) ans += dis[i];
		cout<<ans<<endl; 
	}
	return 0;
}