• 路障【SPFA】


    题目大意:

    求一个无向图的次短路。
    Input

    4 4
    1 2 100
    2 4 200
    2 3 250
    3 4 100
    

    Output

    450

    思路:

    这道题,正解是跑两遍SPFA,一遍是从点1,求出到达其他点的最短路径,记作dis1[i];再从点n,也求出到达其他点的最短路径,记作dis2[i];最后枚举每条边,若两个端点分别是u[i]v[i],长度为dis[i],那么就是求次小的

    dis[i]+dis1[u[i]]+dis2[v[i]]
    dis[i]+dis2[u[i]]+dis1[v[i]]


    代码:

    #include <cstdio>
    #include <iostream>
    #include <queue>
    #define fre(x) freopen(#x".in","r",stdin),freopen(#x".out","w",stdout);
    using namespace std;
    
    const int Inf=99999999;
    int n,m,x,y,z,t,dis1[30001],dis2[30001],vis[30001],head[30001],dis[300001],X[300001],Y[300001],ans,minn,sum;
    
    struct edge  //邻接表
    {
        int next,to,dis;
    }e[300001];
    
    void add(int from,int to,int d)  //连边
    {
        t++;
        e[t].to=to;
        e[t].dis=d;
        e[t].next=head[from];
        head[from]=t;
    }
    
    void spfa1()  //从点1开始跑SPFA
    {
        queue<int> q;
        for (int i=1;i<=n;i++)
        {
            vis[i]=0;
            dis1[i]=Inf;
        }
        vis[1]=1;
        dis1[1]=0;
        q.push(1);
        while (q.size())  //队列不为空
        {
            int u=q.front();
            vis[u]=0;
            q.pop();
            for (int i=head[u];i;i=e[i].next)
            {
                int v=e[i].to;
                if (dis1[v]>dis1[u]+e[i].dis)  //更新最短路
                {
                    dis1[v]=dis1[u]+e[i].dis;
                    if (!vis[v])
                    {
                        q.push(v);
                        vis[v]=1;
                    }
                }
            }
        }
    }
    
    void spfa2()  //从点n开始跑SPFA
    {
        queue<int> q;
        for (int i=1;i<=n;i++)
        {
            vis[i]=0;
            dis2[i]=Inf;
        }
        vis[n]=1;
        dis2[n]=0;
        q.push(n);
        while (q.size())
        {
            int u=q.front();
            vis[u]=0;
            q.pop();
            for (int i=head[u];i;i=e[i].next)
            {
                int v=e[i].to;
                if (dis2[v]>dis2[u]+e[i].dis)
                {
                    dis2[v]=dis2[u]+e[i].dis;
                    if (!vis[v])
                    {
                        q.push(v);
                        vis[v]=1;
                    }
                }
            }
        }
    }
    
    int main()
    {
        fre(block);
        scanf("%d%d",&n,&m);
        for (int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            dis[i]=z;
            X[i]=x;
            Y[i]=y;  //记录
            add(x,y,z);
            add(y,x,z);  //连边
        }
        spfa1();
        spfa2();
        minn=ans=Inf;
        for (int i=1;i<=m;i++)  //求次短路
        {
            sum=dis[i]+dis1[X[i]]+dis2[Y[i]];
            if (sum<minn)
            {
                ans=minn;
                minn=sum;
            }
            else if (sum<ans&&sum!=minn) ans=sum;
            sum=dis[i]+dis2[X[i]]+dis1[Y[i]];
            if (sum<minn)
            {
                ans=minn;
                minn=sum;
            }
            else if (sum<ans&&sum!=minn) ans=sum;
        }
        printf("%d\n",ans);
        return 0;
    }
  • 相关阅读:
    深度图像的获取原理
    第二章 排序 || 第19节 最短子数组练习题
    第二章 排序 || 第18节 有序矩阵查找练习题
    tensorflow 之tf.nn.depthwise_conv2d and separable_conv2d实现及原理
    tensorflow 之常见模块conv,bn...实现
    机器学习中的训练数据不平衡问题
    一些智力题
    Pytoch 抽取中间层特征方法
    娱乐一下
    java访问ad域
  • 原文地址:https://www.cnblogs.com/hello-tomorrow/p/11998860.html
Copyright © 2020-2023  润新知