最短路_HDU_4725

wa了无数发

/*
无向图，共n个点,点分层，x层-x+1层花费C(双向)，此外还有m条额外的路，求1-n最短
然后告诉你点的层数，那么如何建图(if(layer[u]+1==layer[v]))
Dijkstra超时了
Spfa也超，难道还要在搞
稍微用了一下堆，还是超，似乎是建图问题
如何建图比较巧妙，
点是1-N,然后呢层数也是1-N，我们将层拆成点来连接，即第一层为N+1
额，为何将层拆成点需要拆成两点呢？分成入边跟出边，一个点不行？
似乎是可以的，n+1-2n为层的点，然后层层建边，点点建边，层与点建边，点与相邻的层建边，而且相邻的层都有点才能连
并不能理解拆三层的算法，不过据说拆两层会超？

*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define oo 0x3f3f3f3f
const int maxn = 210000;
using namespace std;
struct node
{
    int u,v,val,next;
};
node e[maxn*100];
int head[maxn], tot;
int layer[maxn];
int n,m,c;
int vv[maxn];
int dis[maxn],vis[maxn];
queue<int>q;
queue<int> que;
void init()
{
    tot = 1;
    memset(head, -1,sizeof(head));
    memset(vv, 0,sizeof(vv));
    memset(dis, 0x3f,sizeof(dis));
}
void add(int u, int v,int val)
{
    e[tot].u = u;
    e[tot].v = v;
    e[tot].val = val;
    e[tot].next = head[u];
    head[u] = tot++;
}
void Scan()
{
    init();
        int u,v,w;
    scanf("%d%d%d",&n,&m,&c);
    for(int i=1;i<=n;i++)
    {
          int num;
          scanf("%d",&num);
          layer[i] = num;//记入点层的情况
          vv[num] = 1;//表示层上有点
    }
    for(int i = 1; i < n; i++)
    {
        if(vv[i]&&vv[i+1])//层层间建边，要都有点才能连(层上的点能去层上的点)
        {
            add(n+i,n+i+1,c);
            add(n+i+1,n+i,c);
        }
    }
    for(int i = 1; i <= n; i++)//层与点建边,点与相邻层建边
    {
        add(n+layer[i],i,0);
        if(layer[i]>1)
            add(i,n+layer[i]-1,c);
        if(layer[i]<n)
            add(i,n+layer[i]+1,c);
    }
        for(int i=0;i<m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
}
void Spfa1(int s)
{
     int i,j,u,v,w;
    memset(vis,0,sizeof(vis));
    while(!q.empty()) q.pop();
    dis[1]=0;
    vis[1]=1;
    q.push(1);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        vis[u]=0;
        for(i=head[u]; i!=-1; i=e[i].next)
        {
            v=e[i].v;
            w=e[i].val;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
}
void Spfa(int s)
{
    memset(vis, 0,sizeof(vis));
    while(!que.empty())que.pop();
    dis[s] = 0;
    vis[s] = 1;
    que.push(s);
    while (!que.empty())
    {
        int u = que.front();
        que.pop();
        vis[u] = 0;
        for(int i = head[u]; i!=-1; i=e[i].next)
        {
            int v = e[i].v;
            int w=e[i].val;
            if(dis[v]>dis[u]+w)
            {
                dis[v] = dis[u]+w;
                if(!vis[v])
                {
                    vis[v] = 1;
                    que.push(v);
                }
            }
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    for(int cas = 1; cas<=t;cas++)
    {
        Scan();
        printf("Case #%d: ",cas);
        Spfa(1);
        if(dis[n]!=oo)
        printf("%d\n",dis[n]);
    else
        printf("-1\n");
    }
    return 0;
}