acm_图论_最短路

关键词：最短路、枚举特殊路径

无向路径，已知源点和汇点。路分两种，A路径可任意走，B路径仅能通过一条。求源点到汇点的最短路。

做法：在A路径图中求源点到任意一点的最短路，和任意一点到汇点的最短路，由于路是无向的，所以只要分别求源点和汇点到任意一点的最短路径即可。dij算法恰好提供了求从某点出发到达图上任意一点的最短路径的方法。

设S集合为已知的最短路集合，集合中每个元素由{u,d}组成。u是点编号，d是源点到达u的最短距离。一开始只有点{s,0}。先找到一个离s最近的点u0，则{u0,d<s,u0>}一定属于集合S。这是显然的，可以用反证法证明。用归纳法证明这种贪心选取的正确性。

假设现在已经形成了一个集合S，其中所有点都已达到最短路（显然都是通过S集合中的点到达的）。而不在S集合中的点中，距离值d最小的那个点u此时的d值一定是该点的最短路径。反证法：如果源点到u的最短路径中存在不属于S集合中的点v，则此时v的d值比u的d值小，与u的定义矛盾！因此，u可以加入S集合，由归纳原理，贪心选择正确。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<vector>
#include<stack>
#include<queue>
#define ll long long
#define INF 0x3f3f3f3f
#define maxn 510
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;

int cas,n,m,k,s,t;
int ds[maxn],dt[maxn],vis[maxn];
int parents[maxn],parentt[maxn];
struct Heapnode{//从s到u的距离为d
    int u,d;
    bool operator< (const Heapnode&rhs)const{
        return d>rhs.d;
    }
};
struct Edge{
    int to,d;
};
vector<Edge> g[maxn];

void add(int u,int v,int w){ g[u].push_back((Edge){v,w}); }

void build(){
    scanf("%d",&m);
    for(int i=1;i<=n;i++) g[i].clear();//清零！！！
    for(int i=1;i<=m;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
}

void dij(int s,int d[],int parent[]){
    priority_queue<Heapnode> que;
    for(int i=1;i<=n;i++) d[i]=INF;
    d[s]=0,mem(vis,0),mem(parent,-1);
    que.push((Heapnode){s,0});
    while(!que.empty()){
        Heapnode x=que.top();que.pop();
        int u=x.u;
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=0;i<g[u].size();i++){
            int v=g[u][i].to,w=g[u][i].d;
            if(d[v]>d[u]+w){
                d[v]=d[u]+w;
                parent[v]=u;
                que.push((Heapnode{v,d[v]}));
            }

        }
    }
}

void prints(int s,int v){
    if(s==v) { printf("%d",s);return; }
    prints(s,parents[v]);
    printf(" %d",v);
}

void printt(int u,int t){
    printf(" %d",u);
    if(u==t) return;
    printt(parentt[u],t);
}

void query(){
    scanf("%d",&k);
    int st=0,ed=0,ans=ds[t];
    while(k--){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        if(ds[a]+dt[b]+c<ans) { ans=ds[a]+dt[b]+c,st=a,ed=b; }
        if(ds[b]+dt[a]+c<ans) { ans=ds[b]+dt[a]+c,st=b,ed=a; }
    }
    if(!st){
        prints(s,t);
        printf("\n");
        printf("Ticket Not Used\n");
        printf("%d\n",ans);
    }
    else{
        prints(s,st);
        printt(ed,t);
        printf("\n");
        printf("%d\n",st);
        printf("%d\n",ans);
    }
}

int main(){
    //freopen("a.txt","r",stdin);
    cas=0;
    while(scanf("%d%d%d",&n,&s,&t)!=EOF){
        if(cas) printf("\n");
        cas++;
        build();
        dij(s,ds,parents),dij(t,dt,parentt);
        query();
    }
}