hdu 4725 The Shortest Path in Nya Graph

时间:2023-02-02 21:02:11

题意:给出一个图,求从1到n的最短路,另外,每个结点属于一个层,层数相差为1的两个结点可以相互传送,距离固定为c。

思路:基本算法就是最短路,不管是dijkstra或者spfa都行吧, 我用的是djkstra……主要问题就是构图,当时做热身赛的时候死活没想出来,我果然还是太嫩了么。。。后来洗衣服的时候突然想到一个自以为不错的构图思路,赶紧跑回去写……当时想到的思路是这样的:在相隔为1的两个层之间设一个虚点,然后两个层之间的点分别连向这个虚点,一条边权值为c,另一条为0,结果交完还是wa,后来发现从某一层的点可以通过这个虚点再到达这层的另一个点,知道了这个问题其实就比较好改了,还是这个思路,每一层之间设两个虚点,一个从低一层到高一层,另一个从高一层到低一层,注意这里连的是有向边,这样的话点的个数是n*3,边的个数是n*5,跑一边模板就基本就没问题了~

 

代码:

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
struct Edge
{
    int v,w;
    int next;
};
struct HeapNode
{
    int d,u;
    bool operator < (const HeapNode &a) const
    {
        return d>a.d;
    }
};
Edge edges[maxn*6];
int head[maxn<<2],nEdge,n,m,c;
int d[maxn<<2];
bool vis[maxn<<2];
vector<int>layer[maxn];
void AddEdge(int u,int v,int w)
{
    nEdge++;
    edges[nEdge].v=v;
    edges[nEdge].w=w;
    edges[nEdge].next=head[u];
    head[u]=nEdge;
}
void djkstra(int N)
{
    priority_queue<HeapNode>q;
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=N;++i) d[i]=inf;
    d[1]=0;
    HeapNode hp;
    hp.u=1;hp.d=0;
    q.push(hp);
    while(!q.empty())
    {
        hp=q.top();q.pop();
        int u=hp.u;
        if(vis[u]) continue;
        vis[u]=true;
        for(int k=head[u];k!=-1;k=edges[k].next)
        {
            int v=edges[k].v;
            if(d[v]>d[u]+edges[k].w)
            {
                d[v]=d[u]+edges[k].w;
                hp.d=d[v];hp.u=v;
                q.push(hp);
            }
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t,tcase=0;
    scanf("%d",&t);
    while(t--)
    {
        tcase++;
        int u,v,w,tmp;
        scanf("%d%d%d",&n,&m,&c);
        memset(head,0xff,sizeof(head));
        nEdge=-1;
        for(int i=0;i<=n;++i) layer[i].clear();
        for(int i=1;i<=n;++i)
        {
            scanf("%d",&tmp);
            layer[tmp].push_back(i);
        }
        for(int i=0;i<m;++i)
        {
            scanf("%d%d%d",&u,&v,&w);
            AddEdge(u,v,w);
            AddEdge(v,u,w);
        }
        int sz;
        for(int i=1;i<n;++i)
        {
            sz=layer[i].size();
            for(int j=0;j<sz;++j)
            {
                AddEdge(layer[i][j],i+n,c);
                AddEdge(i+n+n,layer[i][j],0);
            }
            sz=layer[i+1].size();
            for(int j=0;j<sz;++j)
            {
                AddEdge(i+n,layer[i+1][j],0);
                AddEdge(layer[i+1][j],i+n+n,c);
            }
        }
        djkstra(n*3);
        printf("Case #%d: ",tcase);
        if(d[n]==inf||n==0)
          printf("-1\n");
        else printf("%d\n",d[n]);
    }
    return 0;
}