hihocoder1081最短路（dijkstra+堆优化）

题目链接：http://hihocoder.com/problemset/problem/1081

算法主要思路是：将所有节点分为两个集合（已标记节点和未标记节点），已标记节点集合中的所有节点到源节点的最短距离都已经得到。算法主要的工作就是，循环的从未标记节点集合中每次取出一个到源点距离最短的节点放入已标记集合并更新未标记集合中与该节点邻接的节点到源点的距离，直到终节点被标记（单源单宿最短路）或者所有节点都被标记（单源多宿最短路）。

我的代码：

#include <iostream>
#include <vector>
#include <set>
#include <cstring>

using namespace std;

#define MAXN 1005
#define INF 0x7fffffff

struct edge
{
    int y, w;
    edge(int _y, int _w):y(_y),w(_w){}
};

vector<edge> v[MAXN];
int n, m, d[MAXN];
bool vis[MAXN];

struct comp
{
    bool operator()(const int &a, const int &b)const
    {
        if(d[a]==d[b])  return a<b;//这里需要注意一下
        return d[a]<d[b];
    }
};

int dijkstra(int s, int t)
{
    for(int i=1; i<=n; ++i) d[i] = INF;
    d[s] = 0;
    memset(vis+1, 0, n);
    set<int, comp> st;
    for(int i=1; i<=n; ++i) st.insert(i);
    while(!st.empty())
    {
        int x = *st.begin();
        st.erase(x);
        vis[x] = true;
        if(x==t) return d[t];
        int sz = v[x].size();
        for(int i=0; i<sz; ++i)
        {
            int y = v[x][i].y, w = v[x][i].w;
            if(!vis[y]&&d[y]>d[x]+w)
            {
                st.erase(y);
                d[y] = d[x]+w;
                st.insert(y);
            }
        }
    }
}

int main()
{
    int s, t;
    while(cin>>n>>m>>s>>t)
    {
        for(int i=1; i<=n; ++i) v[i].clear();
        while(m--)
        {
            int x, y, w;
            cin>>x>>y>>w;
            v[x].push_back(edge(y, w));
            v[y].push_back(edge(x, w));
        }
        cout<<dijkstra(s, t)<<endl;
    }
    return 0;
}