UVA 11280（SPFA算法变形）

题目大意：

给出n（2<=n<=100)个城市之间的m（0<=m<=1000）条航线以及对应的机票价格，要求回答一些询问，每个询问是给出最大停留次数S，求从其实城市Calgary到终点城市Fredericton中途停留次数不超过s的最便宜的路程。

思路：

SPFA的变形，用二维数组dis[i][j]记录到顶点i步数为j的最短路径。

#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<queue>
#include<map>
#include<vector>
#include<cstdlib>
#include<algorithm>
#define inf 0x3f3f3f3f
#define LL long long
using namespace std;
struct Edge
{
    int from,to,dist;
};
struct node
{
    int id;
    int cnt;
};
vector<Edge> edges;
map<string,int> mapx;
vector<int> g[1015];//存储from对应的边的标号
bool inq[1015][1015];//是否在队列中
int d[1015][1015];
int n,m;//n点的个数，m边的个数
void init()
{
    for(int i=0;i<=n-1;i++)
        g[i].clear();
    edges.clear();
}
void addedge(int from,int to,int dist)//边从0开始
{
    edges.push_back((Edge){from,to,dist});
    int num=edges.size();
    g[from].push_back(num-1);
}
void spfa(int s)
{
    queue<node> q;
    memset(inq,0,sizeof(inq));
    memset(d,0x3f,sizeof(d));
    d[s][0]=0;
    inq[s][0]=1;
    q.push((node){s,0});
    while(!q.empty())
    {
        struct node u=q.front();
        q.pop();
        inq[u.id][u.cnt]=0;
        for(int i=0;i<g[u.id].size();i++)
        {
            Edge e=edges[g[u.id][i]];
            if(d[e.to][u.cnt+1]>d[u.id][u.cnt]+e.dist)
            {
                d[e.to][u.cnt+1]=d[u.id][u.cnt]+e.dist;
                if(!inq[e.to][u.cnt+1])
                {
                    q.push((node){e.to,u.cnt+1});
                    inq[e.to][u.cnt+1]=1;
                }
            }
        }
    }
}
int main()
{
    int T;
    cin>>T;
    int casex=0;
    while(T--)
    {
        casex++;
        cin>>n;
        for(int i=0;i<=n-1;i++)
            {
                string str;
                cin>>str;
                mapx[str]=i;
            }
        cin>>m;
        init();
        for(int i=0;i<=m-1;i++)
        {
            string ax,bx;
            int c;
            cin>>ax>>bx;cin>>c;
            addedge(mapx[ax],mapx[bx],c);
        }
        spfa(0);
        cout<<"Scenario #"<<casex<<endl;
        int t;
        cin>>t;
        while(t--)
        {
            int x;
            cin>>x;
            x=x+1;
            //x=min(x,n-1);
            int ans=inf;
            for(int i=1;i<=x;i++)
                ans=min(ans,d[n-1][i]);
            if(ans==inf)
                cout<<"No satisfactory flights"<<endl;
            else
                cout<<"Total cost of flight(s) is $"<<ans<<endl;
        }
        if(T)
            cout<<endl;
    }
    return 0;
}