首页 > 代码库 > poj 3635 Full Tank 动态规划思想在spfa算法中的应用
poj 3635 Full Tank 动态规划思想在spfa算法中的应用
题意:
有n个城市,和m条已知长度的路,在路上走1单位距离要花1单位油,每个城市都可以加油且有各自的油价,现在任给两个城市s,t,要求从s到t最少花多少油。
思路:
网上大多数都是拿优先队列做的,拿spfa做更有意思,但一般的spfa会超时。dis[i][j]表示到城市i时油箱里有j单位油时的最小花费。对于城市x,y,他们的距离为w,如果存在dis[x][j]<dis[y][i-w],则确定对y进行更新,注意如果对每个dis[x][j]<dis[y][j-w],dis[x][j+1]<dis[y][j-w+1]...都跟新dis[y][m],dis[y][m+1]...dis[y][c]的话会超时,需要用动态规划的思想在确定对y进行更新后只总的更新一次,具体的见代码注释部分。
代码:
#include <iostream> #include <queue> using namespace std; const int maxN=1024; const int maxM=10024; const int maxC=128; struct Edge { int v,w,next; }edge[maxM*2]; int price[maxN]; int head[maxN]; int inq[maxN]; int dis[maxN][maxC]; int n,m,e; void spfa(int c,int s,int t){ int i,j; queue<int> Q; memset(inq,0,sizeof(inq)); for(i=0;i<n;++i) for(j=0;j<=c;++j) dis[i][j]=INT_MAX; for(j=0;j<=c;++j) dis[s][j]=price[s]*j; inq[s]=1; Q.push(s); while(!Q.empty()){ int u=Q.front(); Q.pop(); inq[u]=0; for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].v,w=edge[i].w; int flag=0; for(int j=w;j<=c;++j) if(dis[u][j]<dis[v][j-w]){ dis[v][j-w]=dis[u][j]; flag=1; } if(flag==1){ for(int j=1;j<=c;++j)//用动态规划的思想只更新一次 dis[v][j]=min(dis[v][j],dis[v][j-1]+price[v]); if(inq[v]==0){ inq[v]=1; Q.push(v); } } } /*超时代码: for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].v,w=edge[i].w; int flag=0; for(int j=w;j<=c;++j) if(dis[u][j]<dis[v][j-w]){ dis[v][j-w]=dis[u][j]; for(int k=j-w;k<=c;++k)//每次都尝试到更新到加满 dis[v][k]=min(dis[v][k],dis[v][j-w]+(k-(j-w))*price[v]); flag=1; } if(flag==1&&inq[v]==0){ inq[v]=1; Q.push(v); } } */ } int ans=INT_MAX; for(i=0;i<=c;++i) ans=min(ans,dis[t][i]); if(ans==INT_MAX) printf("impossible\n"); else printf("%d\n",ans); return ; } int main() { e=0; memset(head,-1,sizeof(head)); scanf("%d%d",&n,&m); int i; for(i=0;i<n;++i) scanf("%d",&price[i]); while(m--){ int u,v,w; scanf("%d%d%d",&u,&v,&w); edge[e].v=v;edge[e].w=w;edge[e].next=head[u]; head[u]=e++; edge[e].v=u;edge[e].w=w;edge[e].next=head[v]; head[v]=e++; } int q,c,s,t; scanf("%d",&q); while(q--){ scanf("%d%d%d",&c,&s,&t); spfa(c,s,t); } return 0; }
poj 3635 Full Tank 动态规划思想在spfa算法中的应用
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。