Dijkstra和SPFA都90分,求助
查看原帖
Dijkstra和SPFA都90分,求助
816204
Light_LE楼主2024/10/5 11:32

都是#3没过

Dijkstra:

#include <bits/stdc++.h>
#define maxn 10003
using namespace std;
struct Edge {
	int v, w, next;
}edge[500003];
struct Data {
	int dis, pos;
	bool operator < (const Data &x) const {
		return x.dis < dis;
	}
};
int n, m, s, head[maxn], vis[maxn], dis[maxn];
int cnt;
priority_queue<Data> q;
void add(int u, int v, int w) {
	cnt++;
	edge[cnt].v = v;
	edge[cnt].w = w;
	edge[cnt].next = head[u];
	head[u] = cnt;
}
void dijkstra() {
	memset(dis, 0x3f, sizeof(dis));
	
	dis[s] = 0;
	q.push((Data){0, s});
	
	while (q.size()) {
		Data f = q.top();
		q.pop();
		int pos = f.pos;
		
		if (vis[pos]) {
			continue;
		}
		vis[pos] = 1;
		
		for (int i = head[pos]; i; i = edge[i].next) {
			int v = edge[i].v, w = edge[i].w;
			if (dis[v] > dis[pos] + w) {
				dis[v] = dis[pos] + w;
				if (vis[v] == 0) {
					q.push((Data){dis[v], v});
				}
			}
		}
	}
}
int main() {
	//freopen("in.txt", "r", stdin);
	ios::sync_with_stdio(false); cin.tie(0), cout.tie(0);
	
	cin >> n >> m >> s;
	for (int i = 0; i < m; i++) {
		int u, v, w;
		cin >> u >> v >> w;
		add(u, v, w);
	}
	
	dijkstra();
	
	for (int i = 1; i <= n; i++) {
		cout << dis[i] << " ";
	}
	return 0;
}

SPFA:

#include <bits/stdc++.h>
#define maxn 10003
using namespace std;
struct Edge {
	int v, w, next;
}edge[500003];
int n, m, s, cnt, dis[maxn], vis[maxn], head[maxn];
queue<int> q;
void add(int u, int v, int w) {
	cnt++;
	edge[cnt].v = v;
	edge[cnt].w = w;
	edge[cnt].next = head[u];
	head[u] = cnt;
}
void spfa() {
	memset(dis, 0x3f, sizeof(dis));
	dis[s] = 0, vis[s] = 1;
	q.push(s);
	
	while (q.size()) {
		int f = q.front();
		q.pop(); 
		vis[f] = 0;
		
		for (int i = head[f]; i; i = edge[i].next) {
			int v = edge[i].v, w = edge[i].w;
			if(dis[v] > dis[f] + w) {
				dis[v] = dis[f] + w;
				if (vis[v] == 0) {
					q.push(v);
					vis[v] = 1;
				}
			}
		}
	}
}
int main() {
	//freopen("in.txt", "r", stdin);
	ios::sync_with_stdio(false); cin.tie(0), cout.tie(0);
		
	cin >> n >> m >> s;
	for (int i = 0; i < m; i++) {
		int u, v, w;
		cin >> u >> v >> w;
		add(u, v, w);
	}
	
	spfa();
	
	for (int i = 1; i <= n; i++) {
		cout << dis[i] << " ";
	}
	return 0;
}
2024/10/5 11:32
加载中...