ch2(搜索与图论)——spfa

1.自然语言描述
spfa 算法是bellman-ford算法的队列优化做法,bellman-ford算法中每次松弛会有很多用问更新点去更新邻接的未更新点的情况发生,浪费时间,所以利用队列每次只用已经更新过的点去更新它的邻接点。
2.代码描述
题目:Acwing.851 spfa 求最短路题目链接

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std;

const int MAXN=1e5+10;

int n,m,h[MAXN],e[MAXN*2],ne[MAXN*2],w[MAXN*2],dist[MAXN],idx;
bool st[MAXN];

void add(int a,int b,int c)
{
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=c;
    h[a]=idx++;
}

int spfa()
{
    memset(dist,0x3f,sizeof(dist));
    dist[1]=0;
    queue<int> q;
    q.push(1);
    st[1]=true;
    
    while(!q.empty()){
        auto t=q.front();
        q.pop();
        
        st[t]=false;
        
        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(j);
                    st[j]=true;
                }
            }
        }
    }
    
    if(dist[n]>=0x3f3f3f3f/2)
        return -1;
    return dist[n];
}

int main(void)
{
    cin>>n>>m;
    
    memset(h,-1,sizeof(h));
    
    for(int i=0;i<m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    
    int ans=spfa();
    if(ans==-1)
        cout<<"impossible"<<endl;
    else
        cout<<ans<<endl;
    
    return 0;
}

题目:Acwing.852 spfa 判负环题目链接

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std;

const int MAXN=1e4+10;

int n,m,h[MAXN],e[2*MAXN],ne[2*MAXN],dist[MAXN],w[2*MAXN],cnt[MAXN],idx;
bool st[MAXN];
//cnt保存到某个点时当前遍历路径长度
void add(int a,int b,int c)
{
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=c;
    h[a]=idx++;
}

bool spfa()
{
	//不需要初始化dist:要解决问题是是否存在负环,不关心路径长度
    queue<int> q;
    for(int i=1;i<=n;i++){//图中可能存在负环是1号点无法到达的,所以把所有点加入队列
        q.push(i);
        st[i]=true;
    }
    
    while(!q.empty()){
        auto t=q.front();
        q.pop();
        
        st[t]=false;
            
        for(int i=h[t];i!=-1;i=ne[i]){
            int j=e[i];
            cnt[j]=cnt[t]+1;
            if(cnt[j]>=n)//遍历路径长度>=n,说明产生负环
                return true;
            
            if(dist[j]>dist[t]+w[i]){//负环上的点就会越更新越小
                dist[j]=dist[t]+w[i];
                if(!st[j]){
                    q.push(j);
                    st[j]=true;
                }
            }
        }
        
    }
    return false;
}

int main(void)
{
    cin>>n>>m;
    
    memset(h,-1,sizeof(h));
    
    for(int i=0;i<m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    
    if(spfa())
        cout<<"Yes"<<endl;
    else
        cout<<"No"<<endl;
    
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是SPFA算法记录路径的C++代码模板: ```c++ #include <iostream> #include <cstring> #include <queue> #include <vector> using namespace std; const int MAXN = 1005; const int INF = 0x3f3f3f3f; struct Edge { int to, w; }; vector<Edge> edges[MAXN]; // 存储图的邻接表 int dist[MAXN]; // 存储源点到各个点的最短距离 int pre[MAXN]; // 存储路径上每个点的前驱节点 bool inQueue[MAXN]; // 标记每个点是否在队列中 void SPFA(int s, int n) { queue<int> q; memset(dist, INF, sizeof(dist)); memset(inQueue, false, sizeof(inQueue)); memset(pre, -1, sizeof(pre)); dist[s] = 0; pre[s] = s; inQueue[s] = true; q.push(s); while (!q.empty()) { int u = q.front(); q.pop(); inQueue[u] = false; for (int i = 0; i < edges[u].size(); i++) { int v = edges[u][i].to; int w = edges[u][i].w; if (dist[v] > dist[u] + w) { dist[v] = dist[u] + w; pre[v] = u; // 更新前驱节点 if (!inQueue[v]) { inQueue[v] = true; q.push(v); } } } } } void printPath(int s, int t) { vector<int> path; for (int i = t; i != s; i = pre[i]) { // 从终点往回找前驱节点 path.push_back(i); } path.push_back(s); reverse(path.begin(), path.end()); // 反转路径,使其从起点到终点 for (int i = 0; i < path.size(); i++) { cout << path[i] << " "; } cout << endl; } int main() { int n, m, s, t; cin >> n >> m >> s >> t; for (int i = 0; i < m; i++) { int u, v, w; cin >> u >> v >> w; edges[u].push_back({v, w}); edges[v].push_back({u, w}); // 无向图 } SPFA(s, n); if (dist[t] == INF) { cout << "No path!" << endl; } else { cout << "Shortest path: " << dist[t] << endl; cout << "Path: "; printPath(s, t); } return 0; } ``` 在SPFA算法中,每个节点的前驱节点存储在pre数组中。当算法执行完毕后,可以使用pre数组从终点往回找前驱节点,直到找到起点为止,即可得到最短路径。这里使用一个vector来存储路径上的节点,最后反转vector中的元素,使其从起点到终点输出。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值