洛谷_3371 单源最短路径(弱化版)
程序员文章站
2024-01-14 17:14:16
...
题意
给出一个有向图,求从一个点到其它每一个点的距离。
思路
因为比赛有一题要用SPFA,所以我来用SPFA做一下这题。我们先做一次spfa,然后从1到n输出从起点到点i的距离就好了。
代码
#include<algorithm>
#include<queue>
#include<cstdio>
using namespace std;
int n,m,s,tot,dis[10005],head[500005];
bool vis[10005];
struct Edge{
int next,to,w;
}f[500005];
void add(int u,int v,int w)//用邻接表建图
{
f[++tot].next=head[u];//指向从u点出发的下一条边
f[tot].to=v;//指向第二个点
f[tot].w=w;//记录权值
head[u]=tot;//记录最后一次出现的位置
}
queue<int> q;//stl库的队列
void spfa()
{
for(int i=1; i<=n; i++)
dis[i]=2147483647;//初始化距离为一个很大的数
int u,v;
q.push(s);//让起点先入队
dis[s]=0;//初始化
vis[s]=1;//v[s]表示节点s是否在队列里
while(!q.empty())//如果队列不为空
{
u=q.front();//取队首
q.pop();//相当于head+1,就是让队首出队
vis[u]=0;//要释放标记,因为有可能再次进入队列
for(int i=head[u];i;i=f[i].next)//枚举起点为u
{ //的边
v=f[i].to;//v为这条边的第二个点
if(dis[v]>dis[u]+f[i].w)//松弛操作,这里和
{ //就是更新最短距离
dis[v]=dis[u]+f[i].w;
if(!vis[v])
{
vis[v]=1;//记录
q.push(v);//入队
}
}
}
}
}
int main()
{
scanf("%d%d%d",&n,&m,&s);
for(int i=1;i<=m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);//建图
}
spfa();
for(int i=1;i<=n;i++)
printf("%d ",dis[i]);//输出u到i的最短距离
}
上一篇: C语言练习8—判断回文字符串
下一篇: LCA之ST算法模板