USACO2014FebruaryGold Roadblock Dijkstra+路径还原

先用一遍Dijkstra 然后路径还原 记录一下哪些边处在最短路径上 然后for过去把这几条边都乘个2试试看 再Dijkstra更新答案 比较水的题目 很好想 就是路径还原部分要注意一下 详细可以参见代码

#include<stdio.h>
#include<string.h>
#include<queue>
#include<vector>
#include<iostream>
#include<algorithm>
#define M 255
#define oo 1000000000
using namespace std;
struct node{
    int to;
    int v;
    bool operator <(const node &a)const{
        return v>a.v;
    }
};
vector<node>edge[M];
vector<int>pre[M];
priority_queue<node>Q;
int dis[M],used[M];
bool mark[M];
int dist[M];
void Dijkstra(int n,int x,int t){
    int i,j;
    for(i=1;i<=n;i++){
        dis[i]=oo;
        mark[i]=0;
    }
    while(!Q.empty())Q.pop();
    node tmp;
    tmp.to=x,tmp.v=0;
    Q.push(tmp);
    dis[x]=0;
    while(!Q.empty()){
        tmp=Q.top();
        Q.pop();
        if(mark[tmp.to])continue;
        int k=tmp.to;
        if(k==t)return;
        mark[k]=1;
        for(i=0;i<edge[k].size();i++){
            node nxt=edge[k][i];
            int y=nxt.to;
            if(mark[y])continue;
            if(dis[y]>=dis[k]+nxt.v){
                if(dis[y]>dis[k]+nxt.v)pre[y].clear();//如果不在最短路上就把前驱删去
                pre[y].push_back(k);//路径还原
                dis[y]=dis[k]+nxt.v;
                tmp.to=y;
                tmp.v=dis[y];
                Q.push(tmp);
            }
        }
    }
}
void dfs(int x){
    if(used[x])return;
    used[x]=1;
    for(int i=0;i<pre[x].size();i++)dfs(pre[x][i]);
}
int main(){
    int n,m;
    int i,j,k,x,y,z;
    scanf("%d %d",&n,&m);
    for(i=1;i<=n;i++)edge[i].clear();
    while(m--){
        scanf("%d %d %d",&x,&y,&z);
            node tmp;
            tmp.to=y,tmp.v=z;
            edge[x].push_back(tmp);
            tmp.to=x;
            edge[y].push_back(tmp);
    }
    Dijkstra(n,1,n);
    dfs(n);
    int mi=dis[n];
    int mx=dis[n];
    for(int i=1;i<=n;i++){
        if(!used[i])continue;
        for(int j=0;j<edge[i].size();j++){
            if(!used[edge[i][j].to])continue;
            int to=edge[i][j].to;
            edge[i][j].v*=2;
            for(int k=0;k<edge[to].size();k++){
                if(edge[to][k].to==i){
                    edge[to][k].v*=2;
                    break;
                }
            }
            Dijkstra(n,1,n);
            edge[i][j].v/=2;
            for(int k=0;k<edge[to].size();k++){
                if(edge[to][k].to==i){
                    edge[to][k].v/=2;
                    break;
                }
            }
            mx=max(dis[n],mx);
        }
    }
    printf("%d",mx-mi);
    return 0;
}