洛谷1346 电车 floyed dijkstra

一:floyed(因为点比较少)

#include <bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define res register int
const int maxn=105;
int dp[maxn][maxn];
int N,beg,end;
void floyed(){
	for(res k=1;k<=N;k++)
		for(res i=1;i<=N;i++)
			for(res j=1;j<=N;j++)
				dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
}
void init(){
	memset(dp,inf,sizeof(dp));
	scanf("%d%d%d",&N,&beg,&end);
	int cnt,to;
	for(int i=1;i<=N;i++){
		scanf("%d",&cnt);
		for(int j=1;j<=cnt;j++){
			scanf("%d",&to);
			if(1==j) dp[i][to]=0;
			else dp[i][to]=1;
		}
	}
}

int main()
{
	init();
	floyed();
	if(inf==dp[beg][end]) printf("-1\n");
	else printf("%d\n",dp[beg][end]);
	return 0;
}

 二:Dijkastra 

#include <bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define res register int
const int maxn=105;
int N,beg,end,ans;
int vis[maxn];
int dis[maxn];
int head[maxn];
struct Node{
	int num,sum;
};
bool operator<(const Node x,const Node y){
	return x.sum>y.sum;
}
struct edge{
	int to,next,val;
}e[maxn*maxn];
priority_queue<Node> q;
void add(int x,int y,int c){
	e[ans].to=y;
	e[ans].val=c;
	e[ans].next=head[x];
	head[x]=ans++;
}
void dijsktra()
{
	q.push((Node){beg,0});
	while(!q.empty()){
		Node now=q.top();
		q.pop();
		vis[now.num]=1;
		int x=now.num,y;
		for(res i=head[now.num];i;i=e[i].next){
			y=e[i].to;
			if(dis[y]>dis[x]+e[i].val){
				dis[y]=dis[x]+e[i].val;
				if(!vis[y]) q.push((Node){y,dis[y]}); 
			}
		}
	}
} 

void init()
{
	memset(dis,inf,sizeof(dis));
	scanf("%d%d%d",&N,&beg,&end);
	int cnt,to;
	int v;
	for(int i=1;i<=N;i++){
		scanf("%d",&cnt);
		for(int j=1;j<=cnt;j++){
			scanf("%d",&to);
			if(1==j) v=0;
			else v=1;
			add(i,to,v);
		}
	}
	dis[beg]=0;
}

int main()
{
	init();
	dijsktra();
	if(inf==dis[end]) printf("-1\n");
	else printf("%d\n",dis[end]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41755258/article/details/84872704
今日推荐