一个人的旅行( 链式前向星+(dijstra/spfa) )

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
inline void read(int &x){
    int ans=0,f=1;
    char c=getchar();
    for(;!isdigit(c);c=getchar()){
        if(c=='-')
            f=-1;
    }
    for(;isdigit(c);c=getchar()){
        ans=ans*10+c-'0';
    }
    x=ans*f;
}
typedef struct Edge{
    int to,next,w;
}Edge;
Edge edge[N];
int cnt,head[N],minz=0x3f3f3f3f,dis[N];
int T,S,D;
int s[N],d[N];
void add_edge(int u,int v,int w){
    edge[cnt].to=v;
    edge[cnt].w=w;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
typedef struct Cmp{
    bool operator () (const int &p1,const int &p2){
        return dis[p1]<dis[p2];
    }
}Cmp;
void dijstra(int k){
    memset(dis,127,sizeof(dis));
    dis[k]=0;
    priority_queue< int ,vector<int> , Cmp > q;
    q.push(k);
    while(!q.empty()){
        int u=q.top();
        q.pop();
        for(int i=head[u];~i;i=edge[i].next){
            int v=edge[i].to;
            if(dis[v]>dis[u]+edge[i].w){
                dis[v]=dis[u]+edge[i].w;
                q.push(v);
            }
        }
    }
    for(int i=0;i<S;i++){
        if(minz>dis[s[i]]){
            minz=dis[s[i]];
        }
    }
}
int main()
{
    while(~scanf("%d%d%d",&T,&S,&D)){
          memset(head,-1,sizeof(head));
          for(int i=0;i<T;i++){
                int a,b,c;
                read(a),read(b),read(c);
                add_edge(a,b,c);
                add_edge(b,a,c);
          }
          for(int i=0;i<S;i++){
                read(s[i]);
          }
          for(int i=0;i<D;i++){
                read(d[i]);
          }
          minz=0x3f3f3f3f;
          for(int i=0;i<D;i++){
                dijstra(d[i]);
          }
          printf("%d\n",minz);
    }
    return 0;
}

 SPFA:

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
inline void read(int &ans){
    int f=1,t=0;
    char c=getchar();
    for(;!isdigit(c);c=getchar()){
        if(c=='-') f=-1;
    }
    for(;isdigit(c);c=getchar()){
        t=t*10+c-'0';
    }
    ans=t*f;
}
typedef struct Edge{
    int to,next,w;
}Edge;
Edge edge[N*2];
int head[N],cnt,dis[N],T,S,D,s[1005],d[1005],vis[N];
void add_edge(int u,int v,int w){
    edge[cnt]=Edge{ v , head[u] , w};
    head[u]=cnt++;
}
void spfa(int S){   //八字真言:跑点入队,出队换边
    memset(dis,127,sizeof(dis));
    memset(vis,0,sizeof(vis));
    queue<int>Q;
    Q.push(S);
    vis[S]=1;
    dis[S]=0;
    while(!Q.empty()){
         int u=Q.front();
         Q.pop();
         vis[u]=0;           //出了队列
         for(int i=head[u];~i;i=edge[i].next){
             int v=edge[i].to;
             int w=edge[i].w;
             if(dis[v]>dis[u]+w){
                    dis[v]=dis[u]+w;
                    if(vis[v]==0){      //当前不再队列中
                        vis[v]=1;
                        Q.push(v);
                    }
             }
         }
    }
}
int main()
{
     while(~scanf("%d%d%d",&T,&S,&D)){
           memset(head,-1,sizeof(head));
           int u,v,w;
           for(int i=1;i<=T;i++){
                read(u),read(v),read(w);
                add_edge(u,v,w);
                add_edge(v,u,w);
           }
           for(int i=1;i<=S;i++){
                read(s[i]);
           }
           for(int i=1;i<=D;i++){
                read(d[i]);
           }
           int ans=0x3f3f3f3f;
           for(int i=1;i<=S;i++){
                spfa(s[i]);
                for(int j=1;j<=D;j++){
                    ans=min(dis[d[j]],ans);
                }
           }
           printf("%d\n",ans);
     }
     return 0;
}

猜你喜欢

转载自blog.csdn.net/Z_sea/article/details/81367665