PAT 1030. Travel Plan (30) 图论

1030. Travel Plan (30)

开始想复杂了，也可能是数据太水。回溯竟然能过。

我的想法是先用dijkstra找到距离dis和花费cost的最优解dd[d]和cc[d]

然后用于回溯剪枝优化

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;

#define inf 0x3f3f3f3f
const int N=505;
int dis[N][N],cost[N][N],n,m,s,d;
int dd[N],cc[N],visit[N],ans[N],cnt=0,goal=0;

void dfs(int x,int dist,int money){
	visit[x]=1;
	ans[cnt++]=x;
	if(dist>dd[d]||money>cc[d]||goal)
		return;
	if(x==d){
		goal=1;return;
	}
	for(int i=0;i<n;i++)
		if(!visit[i]&&dis[x][i]){//回溯
			dfs(i,dist+dis[x][i],money+cost[x][i]);
			if(goal==1) return;
			visit[i]=0;
			cnt--;
		}
}

void dijkstra(){
	visit[s]=1;
	for(int i=0;i<n;i++){//s到i的最小值
		dd[i]=dis[s][i];
		cc[i]=cost[s][i];
	}
	for(int p=1;p<n;p++){//加入其他n-1个点
	    int mindis=inf,mincost,next;
	    for(int i=0;i<n;i++){//找到最近点
		    if(visit[i]) continue;
			if(mindis>dd[i]||(mindis==dd[i]&&mincost>cc[i]))
				mindis=dd[i],mincost=cc[i],next=i;
		}
		visit[next]=1;
		for(int i=0;i<n;i++){//更新最小值
			if(visit[i]) continue;
			if(dd[i]>dd[next]+dis[next][i]||(dd[i]==dd[next]+dis[next][i]&&cc[i]>cc[next]+cost[next][i])){
				dd[i]=dd[next]+dis[next][i],cc[i]=cc[next]+cost[next][i];
			}
		}
	}
	memset(visit,0,sizeof visit);
	dfs(s,0,0);//剪枝找到路径ans
	for(int i=0;i<cnt;i++)
		printf("%d ",ans[i]);
	printf("%d %d\n",dd[d],cc[d]);
}

int main(){
	memset(dis,inf,sizeof dis);
	memset(cost,inf,sizeof cost);
    cin>>n>>m>>s>>d;
	int u,v,dist,c;
	for(int i=0;i<m;i++){
		scanf("%d %d %d %d",&u,&v,&dist,&c);
		dis[u][v]=dis[v][u]=dist;
		cost[u][v]=cost[v][u]=c;
	}
	dijkstra();
	return 0;
}