五月赛- ACMER的出行计划（分层图，dij）

Problem: ACMER的出行计划

Problem Description

众所周知，ACMER经常要到全国各地去参加各种比赛。

每当要出去比赛的时候，我们先要制定一个出行计划，比如说我们是坐飞机还是坐火车，从哪里坐到哪里等等。

现在给你一副路线图，请你制定一个合理的出行计划，由于经费有限，规定最多只能坐k次飞机。

Input

第一行给出一个T，表示有T组数据，之后每组数据第一行分别给出n，m，k（n为点数，分别编号为1-n。m为单向边数）

接下来的m行每行分别给出u，v，w，x（u和v分别为两端点编号，方向为u->v。w为费用。x为类型，0表示火车，1表示飞机）

最后一行给出s，e（s为起点，e为终点）

数据范围：1 <= T <= 10; 1 <= n <= 1000; 1 <= m <= 10000; 0 < k < 10; 1 <= u,v <= n; 1 <= w <= 10000。所有数据都为整数

Output

对于每组数据，输出一行最少费用。

Sample Input

1

4 4 1

1 2 1 1

2 4 1 1

1 3 1 0

3 4 2 0

1 4

Sample Output

3

思路：

其实就是很简单的迪杰斯特拉，看成由有k+1层，每一层都有n个点，相邻两层对应同一个地方的点相差n

假设在第k层时 ，从u出发到v，如果坐火车，则还是在这一层上，如果坐飞机，则到第k+1层上

即，表示火车的连线在平面上，而表示飞机的连线在相邻层之间

因为一共只有k+1层，0，1，……，k 分别表示坐n次飞机，所以不会出现坐飞机次数超出给定范围的情况

然后就是有这么多点，用最基本的迪杰斯特拉 最短路算法，一旦既得点在e,e+n,e+2n……e+kn中，这个点对应的值就是所得的最短路

注意点：

1、最短路的思想还是要反复考量一下，一开始WA到心痛就是因为没静下心来好好理代码的思路

2、STL还是很好用的，STL还是很好用的，STL还是很好用的

3、数组不能开太大 （小于10^8）

4、心态还是要调整好，心态要调整好，心态要调整好，要乐观，暂时研究不出来的题可以放一放，先做别的

#include <bits/stdc++.h>
using namespace std;
const int N=1100;
const int K=11;
const long long INF=1e9;
const int M=10010;
map<int, long long> w[N*K];
vector<int> link[N*K];
int n,m,k;
int s,e;

long long dij(int s)
{
    vector<long long> pre(n*k+n, INF);
    pre[s]=0;
    int u=s;
    while(u%n!=e)
    {
        for(int i=0;i<link[u].size();i++)
        {
            int v=link[u][i];
            if(!w[u].count(v)) continue;
            pre[v]=min(pre[v], pre[u]+w[u][v]);
           // printf("AA%d %d %d\n",u,v,w[u][v]);
        }
        pre[u]=1e18;
        vector<long long>::iterator it;
        it=min_element(pre.begin(), pre.end());
        u=distance(pre.begin(), it);
        //printf("CC%d  %d\n",u,pre[u]);
    }
    return pre[u];
}

int main()
{
    //afreopen("1.txt","r",stdin);
    int t;cin>>t;
    while(t--)
    {
        for(int i=0;i<N*K;i++)  w[i].clear();
        cin>>n>>m>>k;
        //memset(w,-1,sizeof(w));
        for(int i=0;i<m;i++)
        {
            int u,v,W,x;
            cin>>u>>v>>W>>x;
            u--;v--;
            if(x==1)
                for(int j=0;j<k;j++)
                    {
                        link[u+j*n].push_back(v+j*n+n);
                        w[u+j*n][v+j*n+n]=W;
                    }
            else
                for(int j=0;j<=k;j++)
                    {
                        link[u+j*n].push_back(v+j*n);
                        w[u+j*n][v+j*n]=W;
                    }
        }
        cin>>s>>e;
        s--;e--;
        long long ans=dij(s);
        cout<<ans<<endl;
    }
    return 0;
}