07-图6 旅游规划 (25分)

做这个题的收获是利用了一维数组来储存无向图，利用函数来较为方便的获得和设置toll和len

#include<stdio.h>
#define Max 10000
typedef struct {
	int len;
	int toll;
} box;
int n,m,s,d;
int collect[501];
int path[501];
box G[129000];
void swap(int &a,int &b) {
	int temp=a;
	a=b;
	b=temp;
}
int getlen(int i,int j) {
	if(i<j) swap(i,j);
	return G[i*(i+1)/2+j].len;
}
int gettoll(int i,int j) {
	if(i<j) swap(i,j);
	return G[i*(i+1)/2+j].toll;
}
void setlen(int i,int len,int toll) {
	int temps=s;
	if(temps>i) swap(temps,i);
	G[i*(i+1)/2+temps].len=len;
	G[i*(i+1)/2+temps].toll=toll;
}
void Dikistra(int v) {
	collect[v]=1;
	while(1) {
		int min=1000,mini=1000;
		for(int i=0; i<n; i++) {
			if(min>getlen(i,v)&&collect[i]==0) {
				min=getlen(i,v);
				mini=i;
			}
		}
		if(mini==1000) break;
		collect[mini]=1;
		for(int i=0; i<n; i++) {
			if(getlen(mini,i)<Max&&collect[i]==0) {
				if(getlen(v,i)>getlen(v,mini)+getlen(mini,i)||(getlen(v,i)==getlen(v,mini)+getlen(mini,i)&&gettoll(v,i)>gettoll(v,mini)+gettoll(mini,i))) {
					setlen(i,getlen(v,mini)+getlen(mini,i),gettoll(v,mini)+gettoll(mini,i))	;//注意需要更新相同长度但更便宜的路线 
				}
			}
		}
	}
}

int main() {
	scanf("%d%d%d%d",&n,&m,&s,&d);
	int p1,p2,len,toll;
	for(int i=0; i<n*(n+1)/2; i++) G[i].len=G[i].toll=Max;
	for(int i=0; i<m; i++) {
		scanf("%d%d%d%d",&p1,&p2,&len,&toll);
		if(p2>p1) swap(p1,p2);
		G[p1*(p1+1)/2+p2].len=len;
		G[p1*(p1+1)/2+p2].toll=toll;
	}
	Dikistra(s);
	printf("%d %d\n",getlen(s,d),gettoll(s,d));


}