搜索与图论--bellman-ford/spfa(C++)

目录

一、bellman-ford

有边数限制的最短路

输出格式

数据范围

输入样例:

输出样例:

代码展示: 

二、spfa 

1.spfa求最短路

输入格式

输出格式

数据范围

输入样例:

输出样例:

代码展示: 

2.spfa判断负环

输入格式

输出格式

数据范围

输入样例:

输出样例:

代码展示: 


一、bellman-ford

有边数限制的最短路

给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环, 边权可能为负数

请你求出从 1 号点到 n 号点的最多经过 k 条边的最短距离,如果无法从 1 号点走到 n 号点,输出 impossible

注意:图中可能 存在负权回路 。

输入格式

第一行包含三个整数 n,m,k。

接下来 m 行,每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。

点的编号为 1∼n。

输出格式

输出一个整数,表示从 1 号点到 n 号点的最多经过 k 条边的最短距离。

如果不存在满足条件的路径,则输出 impossible

数据范围

1≤n,k≤500
1≤m≤10000
1≤x,y≤n
任意边长的绝对值不超过 10000。

输入样例:
3 3 1
1 2 1
2 3 1
1 3 3
输出样例:
3
代码展示: 
#include<bits/stdc++.h>
using namespace std;
const int N = 510 , M = 10010;
struct Edge{
	int a;
	int b;
	int w;
}e[M];
int dist[N];
int back[N];
int n,m,k;
int bellman_ford()
{
	memset(dist,0x3f,sizeof(dist));
	dist[1]=0;
	for(int i=0;i<k;i++)
	{
		memcpy(back,dist,sizeof(dist));
		for(int j=0;j<m;j++)
		{
			int a=e[j].a,b=e[j].b,w=e[j].w;
			dist[b]=min(dist[b],back[a]+w);
		}
	}
	if(dist[n]>0x3f3f3f3f/2)
	return -2;
	else
	return dist[n];
}
int main()
{
	cin>>n>>m>>k;
	for(int i=0;i<m;i++)
	{
		int a,b,w;
		cin>>a>>b>>w;
		e[i]={a,b,w};
	}
	int res=bellman_ford();
	if(res==-2)
	cout<<"impossible";
	else
	cout<<res;
    return 0;
}

二、spfa 

1.spfa求最短路

给定一个 n 个点 m条边的有向图,图中可能存在重边和自环, 边权可能为负数

请你求出 1 号点到 n 号点的最短距离,如果无法从 1 号点走到 n 号点,则输出 impossible

数据保证不存在负权回路。

输入格式

第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。

输出格式

输出一个整数,表示 1 号点到 n 号点的最短距离。

如果路径不存在,则输出 impossible

数据范围

1≤n,m≤10^5
图中涉及边长绝对值均不超过 10000。

输入样例:
3 3
1 2 5
2 3 -3
1 3 4
输出样例:
2
代码展示: 
#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
typedef pair<int,int> PII;
int h[N],e[N],w[N],ne[N],idx;
int st[N];
int dist[N];
int n,m;
void add(int a,int b,int c)
{
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int spfa()
{
	queue<PII> q;
	memset(dist,0x3f,sizeof(dist));
	dist[1]=0;
	q.push({0,1});
	st[1]=1;
	while(!q.empty())
	{
		PII p=q.front();
		q.pop();
		int t=p.second;
		st[t]=0;
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dist[t]+w[i])
			{
				dist[j]=dist[t]+w[i];
				if(!st[j])
				{
					q.push({dist[j],j});
					st[j]=1;
				}
			}
		}
	}
	if(dist[n]==0x3f3f3f3f)
	return -2;
	else
	return dist[n];
}
int main()
{
	cin>>n>>m;
	memset(h,-1,sizeof(h));
	for(int i=0;i<m;i++)
	{
		int a,b,w;
		cin>>a>>b>>w;
		add(a,b,w);
	}
	int res=spfa();
	if(res==-2)
	cout<<"impossible";
	else
	cout<<res;
	return 0;
}

2.spfa判断负环

给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环, 边权可能为负数

请你判断图中是否存在负权回路。

输入格式

第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。

输出格式

如果图中存在负权回路,则输出 Yes,否则输出 No

数据范围

1≤n≤2000,
1≤m≤10000,
图中涉及边长绝对值均不超过 10000。

输入样例:
3 3
1 2 -1
2 3 4
3 1 -4
输出样例:
Yes
代码展示: 
#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int h[N],e[N],w[N],ne[N],idx;
int st[N];
int dist[N];
int cnt[N];
int n,m;
void add(int a,int b,int c)
{
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int spfa()
{
	queue<int> q;
	for(int i=1;i<=n;i++)
	{
		q.push(i);
		st[i]=1;
	}
	st[1]=1;
	while(!q.empty())
	{
		int t=q.front();
		q.pop();
		st[t]=0;
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dist[t]+w[i])
			{
				dist[j]=dist[t]+w[i];
				cnt[j]=cnt[t]+1;
				if(cnt[j]>=n)
				return 1;
				if(!st[j])
				{
					q.push(j);
					st[j]=1;
				}
			}
		}
	}
	return 0;
}
int main()
{
	cin>>n>>m;
	memset(h,-1,sizeof(h));
	for(int i=0;i<m;i++)
	{
		int a,b,w;
		cin>>a>>b>>w;
		add(a,b,w);
	}
	if(spfa())
	cout<<"Yes";
	else
	cout<<"No";
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值