P4822 [BJWC2012]冻结

凌夜绝夕 · 2020-05-15 21:37:54 · 个人记录

最短路分层图模板题

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
using namespace std;
const int N=2600,M=11000000;
int h[N],e[M],ne[M],w[M],idx;
int dist[N];
bool st[N];
int n,m,k;
void add(int a,int b,int c)
{
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=c;
    h[a]=idx++;
}
void spfa()
{
    queue<int> q;
    q.push(1);
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    st[1]=true;
    while(q.size())
    {
        int t=q.front();
        q.pop();
        st[t]=false;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[t]+w[i])
            {
                dist[j]=dist[t]+w[i];
                if(!st[j])
                {
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    memset(h,-1,sizeof h);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
        for(int i=1;i<=k;i++)
        {
            add(a+(i-1)*n,b+i*n,c/2);
            add(b+(i-1)*n,a+i*n,c/2);
            add(a+i*n,b+i*n,c);
            add(b+i*n,a+i*n,c);
        }
    }
    for(int i=1;i<=k;i++)   add(n+(i-1)*n,n+i*n,0);
    spfa();
    printf("%d\n",dist[(k+1)*n]);
    return 0;
}