dijkstra堆优化样例没过求调 玄关
查看原帖
dijkstra堆优化样例没过求调 玄关
879242
SJZ08楼主2024/5/12 22:37

错误代码:

#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int N=1005,M=1000005;
typedef pair<int,int> PII;
priority_queue<PII,vector<PII>,greater<PII> >que;
int h[N],e[M],ne[M],w[M],d[N],v[N],ans[N],idx;
void add(int x,int y,int z) {
	e[idx]=y,w[idx]=z,ne[idx]=h[x],h[x]=idx++;
}
int main() {
	memset(h,-1,sizeof h);
	int n;
	scanf("%d",&n);
	for (int i=0;i<n;i++) {
		scanf("%d",&d[i]);
		que.push({d[i],i});
		ans[i]=1;
	}
	int x,y,z;
	while (scanf("%d %d %d",&x,&y,&z)!=EOF) {
		add(x,y,z);
		if (x!=y) add(y,x,z);
	}
	while (!que.empty()) {
		PII t=que.top();
		que.pop();
		int x=t.second;
		if (v[x]==1) continue;
		v[x]=1;
		for (int i=h[x];i!=-1;i=ne[i]) {
			int j=e[i],k=w[i];
			if (d[k]>d[x]+d[j]) {
				d[k]=d[x]+d[j];
				ans[k]=ans[x]*ans[j];
				if (v[k]==0) {
					que.push({d[k],k});
				}
			}else if (d[k]==d[x]+d[j]) {
				ans[k]+=ans[x]*ans[j];
			}
		}
	}
	printf("%d %d",d[0],ans[0]);
	return 0;
}
2024/5/12 22:37
加载中...