洛谷P1346 电车 dijkstra求最短路

一道很简单的求最短路的题目,不用按开关就能走的,花费cost为0,要按开关才能走的,花费cost为1,dijkstra求最短路即可,下贴代码。

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e4+5;
const int inf=1e9+7;
struct node
{
    int next,to,cost;
}p[maxn];
int n,m,a,b,cnt,t;
int head[105],vertex[maxn];
bool vis[maxn];
void add(int x,int y,int w)
{
    p[++cnt].next=head[x];
    p[cnt].to=y;
    p[cnt].cost=w;
    head[x]=cnt;
}
int dijkstra()
{
    for(int i=1;i<=n;i++)
    vertex[i]=inf;
    vertex[a]=0;
    vis[a]=true;
    for(int i=head[a];i;i=p[i].next)
    {
        int v=p[i].to;
        int c=p[i].cost;
        if(vertex[v]>(c+vertex[a]))
        vertex[v]=c+vertex[a];
    }
    for(int i=1;i<n;i++)
    {
        int Mi=inf,temp;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&vertex[j]<Mi)
            Mi=vertex[j],temp=j;
        }
        if(Mi==inf)
        break;
        vis[temp]=true;
        for(int j=head[temp];j;j=p[j].next)
        {
            int v=p[j].to;
            int c=p[j].cost;
            if(vertex[v]>(Mi+c))
            vertex[v]=Mi+c;
        }
    }
    if(vertex[b]==inf)
    return -1;
    return vertex[b];
}
int main()
{
    scanf("%d %d %d",&n,&a,&b);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&m);
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&t);
            if(j==1)
            add(i,t,0);
            else
            add(i,t,1);
        }
    }
    printf("%d\n",dijkstra());
    return 0;
}
洛谷P1346 电车 dijkstra求最短路洛谷P1346 电车 dijkstra求最短路 烟雨落叶 发布了4 篇原创文章 · 获赞 2 · 访问量 38 私信 关注
上一篇:图的广度遍历(湖北汽车工业学院数据结构实验)


下一篇:图11——判断图中是否为一棵树