PAT刷题——图的最短路径(Dijstra)

PAT1003

Dijstra

#include <bits/stdc++.h>
using namespace std;
const int MAXN=510;
int n,m,st,ed,G[MAXN][MAXN],weight[MAXN];
int d[MAXN],w[MAXN],num[MAXN];
bool vis[MAXN];
const int INF=1000000000;

void dijstra(int s){
	fill(d,d+MAXN,INF);
	memset(num,0,sizeof(num));
	memset(w,0,sizeof(w));
	d[s]=0;
	w[s]=weight[s];
	num[s]=1;
	for(int i=0;i<n;i++){
		int u=-1,MIN=INF;
		for(int j=0;j<n;j++){
			if(vis[j]==false&&d[j]<MIN){
				u=j;
				MIN=d[j];
			}
		}
		if(u==-1)return;
		vis[u]=true;
		for(int v=0;v<n;v++){
			if(vis[v]==false&&G[u][v]!=INF){
				if(d[u]+G[u][v]<d[v]){
					d[v]=d[u]+G[u][v];
					w[v]=w[u]+weight[v];
					num[v]=num[u];
				}else if(d[u]+G[u][v]==d[v]){
					if(w[u]+weight[v]>w[v]){
						w[v]=w[u]+weight[v];
					}
					num[v]+=num[u];
				}
			}
		}
	}
}
int main(){
#ifdef ONLINE_JUDGE
#else
	freopen("1.txt","r",stdin);
#endif
	scanf("%d%d%d%d",&n,&m,&st,&ed);
	for(int i=0;i<n;i++){
		scanf("%d",&weight[i]);
	}
	int u,v;
	fill(G[0],G[0]+MAXN*MAXN,INF);
	for(int i=0;i<m;i++){
		scanf("%d%d",&u,&v);
		scanf("%d",&G[u][v]);
		G[v][u]=G[u][v];
	}
	dijstra(st);
	printf("%d %d",num[ed],w[ed]);
	return 0;
}

Dijstra+DFS

#include <bits/stdc++.h>
using namespace std;
int G[505][505],weight[505],d[505],w[505];
int n,m,st,ed,maxw=-1,num=0;
bool vis[505];
vector<int> pre[505];
vector<int> path,tmppath;
void dijstra(int s){
	fill(d,d+505,INT_MAX);
	d[s]=0;
	for(int i=0;i<n;i++){
		int u=-1,MIN=INT_MAX;
		for(int j=0;j<n;j++){
			if(vis[j]==false&&d[j]<MIN){
				u=j;
				MIN=d[j];
			}
		}
		if(u==-1)return;
		vis[u]=true;
		for(int v=0;v<n;v++){
			if(vis[v]==false&&G[u][v]!=INT_MAX){
				if(d[v]>d[u]+G[u][v]){
					d[v]=d[u]+G[u][v];
					pre[v].clear();
					pre[v].push_back(u);
				}else if(d[v]==d[u]+G[u][v]){
					pre[v].push_back(u);
				}
			}
		}
	}
}
void dfs(int v){
	if(v==st){
		tmppath.push_back(v);
		int cnt=0;
		for(int i=tmppath.size()-1;i>=0;i--){
			cnt+=weight[tmppath[i]];
		}
		if(cnt>maxw){
			path=tmppath;
			maxw=cnt;
		}
		num++;
		tmppath.pop_back();
		return;
	}
	tmppath.push_back(v);
	for(int i=0;i<pre[v].size();i++){
		dfs(pre[v][i]);
	}
	tmppath.pop_back();
}
int main(){
#ifdef ONLINE_JUDGE
#else
	freopen("1.txt","r",stdin);
#endif
	scanf("%d%d%d%d",&n,&m,&st,&ed);
	for(int i=0;i<n;i++){
		scanf("%d",&weight[i]);
	}
	fill(G[0],G[0]+505*505,INT_MAX);
	for(int i=0;i<m;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		scanf("%d",&G[u][v]);
		G[v][u]=G[u][v];
	}
	dijstra(st);
	dfs(ed);
	printf("%d %d",num,maxw);
}