电车 洛谷1346 dp

题目描述

在一个神奇的小镇上有着一个特别的电车网络，它由一些路口和轨道组成，每个路口都连接着若干个轨道，每个轨道都通向一个路口（不排除有的观光轨道转一圈后返回路口的可能）。在每个路口，都有一个开关决定着出去的轨道，每个开关都有一个默认的状态，每辆电车行驶到路口之后，只能从开关所指向的轨道出去，如果电车司机想走另一个轨道，他就必须下车切换开关的状态。

为了行驶向目标地点，电车司机不得不经常下车来切换开关，于是，他们想请你写一个程序，计算一辆从路口A到路口B最少需要下车切换几次开关。

分析

对于每一个点，和第一个点（也就是数据中第一个输入的点）的连边的权值为0，除此以外其他边的权值为1。之所以可以这么搞，是因为每个点最多就访问一次，所以每个路口也最多变更一次状态。

code

//dij+堆优化版本
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>

using namespace std;

struct arr{
int x,y,w,next;
}edge[50000];
int ls[10000];
int dis[10000];
int v[10000];
int n,m,s,t;
int edge_m;

struct cmp{
bool operator ()(int a,int b){
return dis[a]>dis[b];
}
};  //优先队列的定义。

void add(int x,int y,int w)
{
edge_m++;
edge[edge_m]=(arr){x,y,w,ls[x]};ls[x]=edge_m;
}

int main()
{
scanf("%d%d%d",&n,&s,&t);
priority_queue<int,vector<int>,cmp> Q;
memset(dis,63,sizeof(dis));
for (int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
for (int j=1;j<=x;j++)
{
int y;
scanf("%d",&y);
if (j==1) add(i,y,0);
else add(i,y,1);
}
}
dis[s]=0; v[s]=1;
Q.push(s);
for (int ii=1;ii<=n;ii++)
{
int x=Q.top();
Q.pop();
for (int j=ls[x];j;j=edge[j].next)
{
if (dis[edge[j].y]>dis[x]+edge[j].w)
{
dis[edge[j].y]=dis[x]+edge[j].w;
if (!v[edge[j].y])
{
v[edge[j].y]=1;
Q.push(edge[j].y);
}
}
}
}
if (dis[0]==dis[t])
printf("-1");
else
printf("%d",dis[t]);
}