单源最短路扩展:创建虚拟源点

解题思路:题目大致意思是求在题目给出的点中离s最近的距离,其实我们可以创建一个虚拟源点让各个点到他的路径是0,spfa 0 就行了。

#include<bits/stdc++.h>
using namespace std;
const   int N=1010,M = 50010;
int e[M],h[N],idx,ne[M],w[M];
int dist[N];
bool st[N];
int n , m , s ;
void add(int a,int b,int c)
{
   
    w[idx] =  c ,ne[idx]=h[a],e[idx]=b ,h[a]= idx++;
}
int spfa()
{
   
    memset(st,0,sizeof st);
    memset(dist,0x3f,sizeof dist);
    dist[0]=0;
    queue<int>q;
    q.push(0);
    st[0]=true;
    while(q.size())
    {
   
        int t=q.front();
        q.pop();
        st[t]=false;
        for(int i=h[t];~i;i=ne[i])
        {
   
            int j=  e[i];
            if(dist[j]>dist[t]+w[i])
            {
   
                dist[j]=dist[t]+w[i];
                if(!st[j])
                {
   
                    st[j]=true;
                    q.push(j);
                }
            }
        }
        
    }
    if(dist[s]==0x3f3f3f3f) return -1;
    return dist[s];

    
}
int main()
{
   
    while(cin >>  n >> m >>s)
    {
   
        idx =  0;
        memset(h,-1,sizeof h);
        while(m--)
        {
   
            int a,b,c;
            cin >> a >> b >>c;
            add(a,b,c);
        }
        int t;
        cin >> t;
        while(t--)
        {
   
            int x;
            cin >>x;
            add(0,x,0);
        }
        cout << spfa()<<endl;
    }
}
全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务