P4568 [JLOI2011]飞行路线

题目链接

本题呢和以前最短路一毛一样，难就难在了建边，说实话我知道是分层图也不怎么会写，我还是太菜了，建图看一下代码应该就会了，最短路就是板子也不说了，我想吐槽的是这一题可以不用用到k次就可以到达终点，而且这个数据还很毒瘤。解决方法就是把终点的每一层都连起来。

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<string>
#include<vector>
#include<queue>
#include<algorithm>
#include<deque>
#include<map>
#include<stdlib.h>
#include<set>
#include<iomanip>
#include<stack>
#define ll long long
#define ms(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x & -x
#define fi first
#define se second
#define bug cout<<"----acac----"<<endl
#define IOS ios::sync_with_stdio(false), cin.tie(0),cout.tie(0)
using namespace std;
const int maxn = 2e6 + 50;
const int maxm = 1.5e5+50;
const double eps = 1e-7;
const double inf = 0x3f3f3f3f;
const ll  lnf  = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9+7;
const  double pi=3.141592653589;
int n,m,k,s0,t0;
int head[maxn],len=0,dis[maxn],vis[maxn];
struct node
{
    int to,next,val;
}a[maxn<<1];
struct point
{
    int id,val;
    point(int id,int val)
    {
        this->id=id;
        this->val=val;
    }
    bool operator<(const point&x)const
    {
        return val>x.val;
    }
};
void add(int u,int v,int w)
{
    a[len].to=v;
    a[len].next=head[u];
    a[len].val=w;
    head[u]=len++;
}
void dij(int s)
{
    priority_queue<point>q;
    ms(vis,0);
    ms(dis,inf);
    dis[s]=0;
    q.push(point(s,0));
    while(!q.empty())
    {
        int u=q.top().id;
        q.pop();
        if(vis[u])continue;
        vis[u]=1;
        for(int i=head[u];~i;i=a[i].next)
        {
            int v=a[i].to;
            if(dis[v]>dis[u]+a[i].val)
            {
                dis[v]=dis[u]+a[i].val;
                q.push(point(v,dis[v]));
            }
        }
    }
}
int main()
{
    ms(head,-1);
    len=0;
    scanf("%d%d%d",&n,&m,&k);
    scanf("%d%d",&s0,&t0);

    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
       // u++,v++;
        add(u,v,w);
        add(v,u,w);
        for(int j=1;j<=k;j++)
        {
            add(u+(j-1)*n,v+j*n,0);
            add(v+(j-1)*n,u+j*n,0);
            add(v+j*n,u+j*n,w);
            add(u+j*n,v+j*n,w);
        }
    }
    for(int i=1;i<=k;i++)
    {
        add(t0+(i-1)*n,t0+i*n,0);
    }
    dij(s0);
    printf("%d\n",dis[t0+k*n]);
    return 0;
}