UVA-10269 (floyd+dijkstra)

题意：

现在有A个村庄,B个城堡,现在要从1到A+B,有M条路,魔法鞋最多能用K次,每次的长度不超过L,且起点和终点一定是村庄和城堡,而且每次使用魔法鞋不能穿过城堡,问最短时间是多少;

思路：

先用Floyd算出不经过城堡的两点之间的最短距离,然后就是加一维的dijkstra了,转移就是看是否使用魔法鞋;

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <bits/stdc++.h>
#include <stack>
#include <map>

using namespace std;

#define For(i,j,n) for(int i=j;i<=n;i++)
#define mst(ss,b) memset(ss,b,sizeof(ss));

typedef  long long LL;

template<class T> void read(T&num) {
    char CH; bool F=false;
    for(CH=getchar();CH<'0'||CH>'9';F= CH=='-',CH=getchar());
    for(num=0;CH>='0'&&CH<='9';num=num*10+CH-'0',CH=getchar());
    F && (num=-num);
}
int stk[70], tp;
template<class T> inline void print(T p) {
    if(!p) { puts("0"); return; }
    while(p) stk[++ tp] = p%10, p/=10;
    while(tp) putchar(stk[tp--] + '0');
    putchar('\n');
}

const LL mod=1e6+3;
const double PI=acos(-1.0);
const int inf=1e9;
const int N=2e5+10;
const int maxn=1e3+520;
const double eps=1e-12;

int A,B,M,L,K,n;
int d[105][105],mp[105][105],dis[105][12],vis[105];
inline void Init()
{
    n=A+B;
    For(i,1,n)
    {
        For(j,1,n)
        {
            if(i==j)mp[i][j]=d[i][j]=0;
            else mp[i][j]=d[i][j]=inf;
        }
    }
    For(i,1,n)
    {
        For(j,0,K)
        {
            if(i==1)dis[i][j]=0;
            else dis[i][j]=inf;
        }
    }
}

void floyd()
{
    for(int k=1;k<=A;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(d[i][j]>d[i][k]+d[k][j])
                    d[i][j]=d[i][k]+d[k][j];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(d[i][j]<=L)d[i][j]=0;
}
inline void solve()
{
    floyd();
    for(int i=1;i<=n;i++)dis[i][0]=mp[1][i];

    for(int k=0;k<=K;k++)
    {
        mst(vis,0);
        for(int i=1;i<=n;i++)
        {
            int mmin=inf,temp=1;
            for(int j=1;j<=n;j++)
            {
                if(!vis[j]&&dis[j][k]<mmin)
                {
                    mmin=dis[j][k];
                    temp=j;
                }
            }
            vis[temp]=1;
            for(int j=1;j<=n;j++)
            {
                if(dis[j][k]>dis[temp][k]+mp[temp][j])dis[j][k]=dis[temp][k]+mp[temp][j];
                if(dis[j][k+1]>dis[temp][k]+d[temp][j])dis[j][k+1]=dis[temp][k]+d[temp][j];
            }
        }
    }
    int ans=inf;
    for(int i=0;i<=K;i++)ans=min(ans,dis[n][i]);
    printf("%d\n",ans);
}
int main()
{
    int t;
    read(t);
    while(t--)
    {
        read(A);read(B);read(M);read(L);read(K);
        Init();
        int u,v,w;
        For(i,1,M)
        {
            read(u);read(v);read(w);
            w=min(w,mp[u][v]);
            mp[u][v]=mp[v][u]=d[u][v]=d[v][u]=w;
        }
        solve();
    }
    return 0;
}