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;
}