二维spfa模板

题目描述

• 编号为1-N的N座城镇用若干仅供单向行驶的道路相连，每条道路上均有两个参数：道路长度(lenth)和在该条道路上行驶的费用（cost）。

          BOB准备从城镇1出发到达城镇N，但他目前只有W的钱，为此，你需要帮助他寻找一条从城镇1到城镇N在他能支付的前提下的一条最短路线。

输入格式

 

• W,N,M（N为城镇数目，2<=N<=100，M为道路条数，W为钱的数目
• 随后的M行每行为一条道路的信息，包含4个数值（u,v,w,cost），表示从城镇u到v有一条长度为cost的单向道路，经过这条道路需要花费 cost。（1<=u,v<=N,1<=w<=100,1<=cost<=100)

输出格式

输出最短长度，若无解，则输出“NO”；

样例

样例输入

5 6 7
1 2 2 3
2 4 3 3
3 4 2 4
1 3 4 1
4 6 2 1
3 5 2 0
5 4 3 2

样例输出

11

#include<bits/stdc++.h>
using namespace std;
const int maxn=100+10,maxm=10000+10,inf=0x3f3f3f3f;
    int w,n,m;
struct Edge{
    int to,next,lenth,money;
}e[maxm];
int head[maxn],tot=0;
void Insert(int a,int b,int c,int d){
    e[++tot].to=b;
    e[tot].lenth=c;
    e[tot].money=d;
    e[tot].next=head[a];
    head[a]=tot;
}
int d[maxn][1000+10];
void spfa(int x){
    queue< pair<int,int> > q;
    int vis[maxn][1000+10];
    memset(d,0x3f,sizeof(d));
    memset(vis,0,sizeof(vis));
    q.push(make_pair(x,0));
    vis[x][0]=1;
    d[x][0]=0;
    while(!q.empty()){
        pair<int,int> now=q.front();
        q.pop();
        int u=now.first;
        int s=now.second;
        vis[u][s]=0;
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].to;
            if(s+e[i].money>w) continue;
            if(d[v][s+e[i].money]>d[u][s]+e[i].lenth){
                d[v][s+e[i].money]=d[u][s]+e[i].lenth;
                if(!vis[v][s+e[i].money]){
                    vis[v][s+e[i].money]=1;
                    q.push(make_pair(v,s+e[i].money));
                }
            }
        }
    }
}
int main(){
    scanf("%d%d%d",&w,&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,l,cost;
        scanf("%d%d%d%d",&u,&v,&l,&cost);
        Insert(u,v,l,cost);
    }
    spfa(1);
    int ans=inf;
    for(int i=0;i<=w;i++){
        ans=min(ans,d[n][i]);
    }
    if(ans==inf) printf("NO
");
    else printf("%d
",ans);
    return 0;
}