tyvj 1031 热浪 最短路

题目链接：http://tyvj.cn/p/1031

背景

USACO OCT09 9TH

描述

德克萨斯纯朴的民眾们这个夏天正在遭受巨大的热浪！！！他们的德克萨斯长角牛吃起来不错，可是他们并不是很擅长生產富含奶油的乳製品。Farmer John此时以先天下之忧而忧，后天下之乐而乐的精神，身先士卒地承担起向德克萨斯运送大量的营养冰凉的牛奶的重任，以减轻德克萨斯人忍受酷暑的痛苦。

FJ已经研究过可以把牛奶从威斯康星运送到德克萨斯州的路线。这些路线包括起始点和终点先一共经过T (1 <= T <= 2,500)个城镇，方便地标号為1到T。除了起点和终点外地每个城镇由两条双向道路连向至少两个其它地城镇。每条道路有一个通过费用（包括油费，过路费等等）。考虑这个有7个城镇的地图。城镇5是奶源，城镇4是终点（括号内的数字是道路的通过费用）。

                              [1]----1---[3]-
                             /               
                      [3]---6---[4]---3--[3]--4
                     /               /       /|
                    5         --[3]--  --[2]- |
                            /        /       |
                      [5]---7---[2]--2---[3]---
                            |       /
                           [1]------

经过路线5-6-3-4总共需要花费3 (5->6) + 4 (6->3) + 3 (3->4) = 10的费用。

给定一个地图，包含C (1 <= C <= 6,200)条直接连接2个城镇的道路。每条道路由道路的起点Rs，终点Re (1 <= Rs <= T; 1 <= Re <= T)，和花费(1 <= Ci <= 1,000)组成。求从起始的城镇Ts (1 <= Ts <= T)到终点的城镇Te(1 <= Te <= T)最小的总费用。

 

分析：最短路。

#include <bits/stdc++.h>

using namespace std;

const int maxn = 2510;
const int inf = 0x3f3f3f3f;

struct Edge {
    int from,to,dist;
};

struct HeapNode {
    int u,d;
    bool operator < (const HeapNode& rhs ) const {
        return d > rhs.d;
    }
};

struct Dijstra {
    int n,m;
    vector<Edge> edges;
    vector<int> G[maxn];
    int d[maxn];
    bool done[maxn];
    int p[maxn];

    void init(int n) {
        this->n = n;
        for(int i=0;i<n;i++)
            G[i].clear();
        edges.clear();
    }

    void AddEdge(int from,int to,int dist) {
        edges.push_back((Edge){from,to,dist});
        int m = edges.size();
        G[from].push_back(m-1);
    }

    void dijstra(int s) {
        memset(done,0,sizeof(done));
        for(int i=0;i<n;i++)
            d[i] = inf;

        d[s] = 0;
        priority_queue<HeapNode> Q;
        Q.push((HeapNode){s,0});

        while(!Q.empty()) {

            HeapNode x = Q.top();
            Q.pop();

            if(done[x.u])
                continue;

            for(int i=0;i<G[x.u].size();i++) {
                Edge& e = edges[G[x.u][i]];
                if(d[e.to]>d[x.u]+e.dist) {
                    d[e.to] = d[x.u] + e.dist;
                    Q.push((HeapNode){e.to,d[e.to]});
                }
            }
        }
    }
};
Dijstra sol;

int main()
{
    int t,c,ts,te;
    scanf("%d%d%d%d",&t,&c,&ts,&te);
    ts--;
    te--;

    sol.init(t);
    for(int i=0;i<c;i++) {
        int u,v,dist;
        scanf("%d%d%d",&u,&v,&dist);
        u--;v--;
        sol.AddEdge(u,v,dist);
        sol.AddEdge(v,u,dist);
    }

    sol.dijstra(ts);
    printf("%d
",sol.d[te]);


    return 0;
}