POJ-3255-Roadblocks-次短路

题意:

求次短路

思路:

利用spfa算法,先求出以1为起点的单源最短路存入d[],再求出以n为起点的单源最短路存入d1[]
若当前存在一条边(u,v)则d[u]+d1[v]+cost[u][v],即为1~n的一条可能路径
遍历所以边,超出一条大于最短路且最短的可能路径,即是次短路

代码:

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=5000+10;
const int maxm=200000+10;
const int inf=0x3f3f3f3f;
int d[maxn],d1[maxn],p[maxn];
int eid=0;
int n,r;
struct Edge
{
    int from,to,cost,next;
}e[maxm];
void init()
{
    memset(p,-1,sizeof(p));
    eid=0;
}
void insert(int u,int v,int c)
{
    e[eid].from=u;
    e[eid].to=v;
    e[eid].cost=c;
    e[eid].next=p[u];
    p[u]=eid++;
}
void addedge(int u,int v,int c)
{
    insert(u,v,c);
    insert(v,u,c);
}
bool inq[maxn];
int cnt[maxn];  //记录每个顶点入队次数,若某点入队超过n次,则存在负环
// 如果到顶点 i 的距离是 0x3f3f3f3f,则说明不存在源点到 i 的最短路
void spfa(int s,int d[]) {
    memset(inq, 0, sizeof(inq));
    for(int i=0;i<=n;i++)
        d[i]=inf;
    d[s] = 0;
    inq[s] = true;
    queue<int> q;
    q.push(s);
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        inq[u] = false;
        for (int i = p[u]; i != -1; i = e[i].next) {
            int v = e[i].to;
            if (d[u] + e[i].cost < d[v]) {
                d[v] = d[u] + e[i].cost;
                if (!inq[v]) {
                    q.push(v);
                    cnt[v]++;
                    inq[v] = true;
                }
            }
        }
    }
}
int main()
{
    while(cin>>n>>r)
    {
        init();
        while(r--)
        {
            int u,v,w;
            cin>>u>>v>>w;
            addedge(u,v,w);
        }
        spfa(1,d);
        spfa(n,d1);
        int ans=d[n],res=inf;
        for(int i=0;i<eid;i++)
        {
            int u=e[i].from,v=e[i].to,w=e[i].cost;
            if(d[u]+d1[v]+w>ans)                 //求一条大于最短路的最短路径
                res=min(res,d[u]+d1[v]+w);
        }
        cout<<res<<endl;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_43093481/article/details/82319143