2018 宁夏省赛 F. Moving On

题目链接 https://nanti.jisuanke.com/t/28406

大意是 有n(<=200)个城市，城市间有路(Input给了邻接矩阵)  每个城市有一个危险值，然后是q(2e4)个询问，每个询问给了 u，v ，w ，对于每个询问回答u到v的最短路长度(最短路过程中不得经过危险值超过w的城市，ps不含首尾)

开始看见邻接矩阵的形式猜了用Floyd写，想的是每次询问跑一次最短路(循环k时避过危险值超过w的点)，但q有2e4，会TLE；后来又想每次询问跑一遍Dijkstra(改造后的不经过危险值超出的点)，以为Dijkstra能快一点，但还是TLE了。

正确的做法还是Floyd，按危险值从小到大排序城市编号(间接排序)，然后用的dp[][][]记录最短路。

dp[k][i][j]表示 添加(第1~k小的危险值的城市后)的 i->j 最短路。(其实就是Floyd三维未降维的版本)

然后针对每个询问找到它的最大k 的最短路图。

#include<bits/stdc++.h>
#define EPS 1e-9
using namespace std;

typedef long long ll;
const int INF=0x3f3f3f3f;
const int MAXN=211;

int rob[211];
int dp[211][211][211];
int id[211];

bool cmp(int i,int j){
    return rob[i]<rob[j];
}

int main(){
    int Tests;
    scanf("%d",&Tests);
    for(int cntT=1;cntT<=Tests;++cntT){
        memset(dp,INF,sizeof(dp));
        int N,Q;
        scanf("%d%d",&N,&Q);
        for(int i=1;i<=N;++i){
            id[i]=i;
            scanf("%d",&rob[i]);
        }
        sort(id+1,id+N+1,cmp);

        for(int i=1;i<=N;++i)
            for(int j=1;j<=N;++j)
                scanf("%d",&dp[0][i][j]);
        for(int k=1;k<=N;++k){
          int rk=id[k];
          for(int i=1;i<=N;++i)
            for(int j=1;j<=N;++j) {
                dp[k][i][j]=min(dp[k-1][i][j],dp[k-1][i][rk]+dp[k-1][rk][j]);
            }
        }
        printf("Case #%d:
",cntT);
        for(int q=1;q<=Q;++q){
            int u,v,dan;
            scanf("%d%d%d",&u,&v,&dan);
            int k=0;
            for(int i=1;i<=N;++i)
                if(rob[id[i]]<=dan) k=i;
            printf("%d
",dp[k][u][v]);
        }
    }
    return 0;
}

主要是熟悉Floyd原型。