POJ 3255 Roadblocks

“ Ctrl AC！一起AC！”

题目：

思路：

用Dijkstra，但是我们每次不仅仅更新最短距离，我们还要更新次短距离。 

代码：

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define LL long long
using namespace std;
typedef pair<int,int> P;
const int MAX=1e9+7;
const int MAXN=1e6+10;
int n,r;
struct edge{
	int to,cost;
};
vector<edge> graph[5010];
int dis1[5010];
int dis2[5010];

void slove()
{
    memset(dis1,inf,sizeof(dis1));
    memset(dis2,inf,sizeof(dis2));
    priority_queue <P,vector<P>,greater<P> > q;
    dis1[1]=0;
    q.push(P(0,1));
    while(!q.empty())
    {
        P p=q.top();
        q.pop();
        int v=p.second,d=p.first;
        if(dis2[v]<d) continue;
        for(int i=0;i<graph[v].size();i++)
        {
            edge e=graph[v][i];
            int d2=d+e.cost;
            if(dis1[e.to]>d2)
            {
                swap(dis1[e.to],d2);
                q.push(P(dis1[e.to],e.to));
            }
            if(dis2[e.to]>d2&&dis1[v]<d2)
            {
                dis2[e.to]=d2;
                q.push(P(dis2[e.to],e.to));
            }
        }
    }
    printf("%d\n",dis2[n]);
}

int main()
{
    while(scanf("%d%d",&n,&r)!=EOF)
    {
        int a,b,d;
        for(int i=0;i<r;i++)
        {
            scanf("%d%d%d",&a,&b,&d);
            graph[a].push_back(edge{b,d});
            graph[b].push_back(edge{a,d});
        }
        slove();
    }
    return 0;
}

感谢阅读！！！

“ Ctrl AC！一起 AC！”