牛客 小米校招 分布式集群消息传递最短路模板题

题目描述
有一个分布式服务集群，集群内含有 N 个服务节点，分别标记为 1 到 N。
给予一个列表 times，表示消息从两个节点间有向传递需要的时间。 times[i] = (s, d, t)，其中 s 表示发出消息的源节点，d 表示接收到消息的目标节点， t 表示信息有向传递的时间。
现在 K 节点发送了一个信号，请问至少需要多少秒才能使所有的服务节点都收到该消息？如果消息不能传递给集群内全部节点，则返回-1。
输入描述:

第一行：列表 times。分布式服务集群的图，图的结构为二维数组。例如： [[2,1,1],[2,3,1],[3,4,1]] ，表示集群4个节点，2到1的时间为1，2到3的时间为1，3到4的时间为1；
第二行：N值
第三行：K值
范围约束：

1. N 的范围在 [1, 100] 之间。
2. K 的范围在 [1, N] 之间。
3. times 的长度在 [1, 6000] 之间。
4. 所有的边 times[i] = (s, d, t) 都有 1 <= s, d <= N 且 1 <= t <= 100。

输出描述:

至少需要多少秒才能使所有的服务节点都收到该消息？如果消息不能传递给集群内全部节点，则返回-1

示例1
输入
复制

[[2,1,1],[2,3,1],[3,4,1]]
4
2

输出
复制

2

备注:

图可能存在重边或自环

题目 : https://www.nowcoder.com/practice/b5029772b0814250a5493b69564e4a04?tpId=125&&tqId=33761&rp=1&ru=/activity/oj&qru=/ta/exam-xiaomi/question-ranking

注意是有向图
迪杰斯特拉模板

int dij(int S) {
	priority_queue<Node> q;
	memset(d, INF, sizeof(d));
	q.push({-d[S], S}), d[S] = 0;
	while(!q.empty()) {
		auto now = q.top(); q.pop();
		int u = now.second;
		for(auto ed : G[u]) {
			int v = ed.to, td = ed.w;
			if(d[u]+td < d[v]) {
				d[v] = d[u] + td;
				q.push({-d[v], v});
			}
		}
	}
	// forarr(d, 1, n);
	int ret = 0;
	for(int i=1; i<=n; i++) ret = max(ret, d[i]);
	return ret;
}

完整代码

#define debug
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <math.h>

#define MAXN ((int)1e5+7)
#define ll long long int
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for(int i=lef; i<=rig; i++)
#define forj(lef, rig) for(int j=lef; j<=rig; j++)
#define fork(lef, rig) for(int k=lef; k<=rig; k++)
#define QAQ (0)

using namespace std;

#define show(x...) \ do { \ cout << "\033[31;1m " << #x << " -> "; \ err(x); \ } while (0)

void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }

namespace FastIO{

	char print_f[105];
	void read() {}
	void print() { putchar('\n'); }

	template <typename T, typename... T2>
		inline void read(T &x, T2 &... oth) {
			x = 0;
			char ch = getchar();
			ll f = 1;
			while (!isdigit(ch)) {
				if (ch == '-') f *= -1; 
				ch = getchar();
			}
			while (isdigit(ch)) {
				x = x * 10 + ch - 48;
				ch = getchar();
			}
			x *= f;
			read(oth...);
		}
	template <typename T, typename... T2>
		inline void print(T x, T2... oth) {
			ll p3=-1;
			if(x<0) putchar('-'), x=-x;
			do{
				print_f[++p3] = x%10 + 48;
			} while(x/=10);
			while(p3>=0) putchar(print_f[p3--]);
			putchar(' ');
			print(oth...);
		}
} // namespace FastIO
using FastIO::print;
using FastIO::read;

int n, m, K;

struct Edge {
	int to, w;
} ;

vector<Edge> G[MAXN];
string line;

inline void add(int u, int v, int w) {
	G[u].push_back({v, w}) /*, G[v].push_back({u, w}) */;
}
#if 1
typedef pair<int, int> Node;
int d[MAXN];
int dij(int S) {
	priority_queue<Node> q;
	memset(d, INF, sizeof(d));
	q.push({-d[S], S}), d[S] = 0;
	while(!q.empty()) {
		auto now = q.top(); q.pop();
		int u = now.second;
		for(auto ed : G[u]) {
			int v = ed.to, td = ed.w;
			if(d[u]+td < d[v]) {
				d[v] = d[u] + td;
				q.push({-d[v], v});
			}
		}
	}
	// forarr(d, 1, n);
	int ret = 0;
	for(int i=1; i<=n; i++) ret = max(ret, d[i]);
	return ret;
}

#else 
int dij(int s) {
	int d[MAXN];
	memset(d, 0x7f7f7f7f, sizeof(d));
	priority_queue<pair<int, int> > Q;
	Q.push(make_pair(-d[s],s)), d[s] = 0;
	while(!Q.empty()) { //dijkstra+堆优化
		int now = Q.top().second;
		Q.pop();
		for(int i=0; i<(int)G[now].size(); i++) {
			int v = G[now][i].to;
			if(d[v] > d[now]+G[now][i].w) { //松弛
				d[v] = d[now] + G[now][i].w;
				Q.push(make_pair(-d[v], v));
			}
		}
	}
	int ret = -99999;
	for(int i=1; i<=n; i++) ret = max(ret, d[i]);
	return ret;
}
#endif

int main() {
#ifdef debug
	freopen("test", "r", stdin);
	clock_t stime = clock();
#endif
	int u = 0, v = 0, w = 0;
#if 1
	char ch;
	while((ch=getchar()) && ch!=']') {
		scanf("%c%d%c%d%c%d%c", &ch, &u, &ch, &v, &ch, &w, &ch);
		add(u, v, w);
	}
#else 
	char ch = getchar(), s;
	while((ch=getchar()) == '[') {
		scanf("%d%c%d%c%d%c%c", &u, &s, &v, &s, &w, &s, &s);
		add(u, v, w);
	}
#endif
	int S;
	cin >> n >> S;
	int ans = dij(S);
	printf("%d\n", ans==0x7f7f7f7f ? -1 : ans);


#ifdef debug
	clock_t etime = clock();
	printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
#endif 
	return 0;
}