一道很简单的求最短路的题目,不用按开关就能走的,花费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;
}
烟雨落叶
发布了4 篇原创文章 · 获赞 2 · 访问量 38
私信
关注