洛谷P1339 热浪【最短路】

题目：https://www.luogu.org/problemnew/show/P1339

题意：给定一张图，问起点到终点的最短路。

思路：dijkstra板子题。

很久没有写最短路了。总结一下dijkstra的步骤吧。

d数组用于表示当前最短路径，vis数组用于标记当前点是否已经在最短路集合中了。

每次找到一个d最小的节点，表示他已经无法更短了，把他加入集合，用他去更新其他的节点。一共做n-1次。

#include<cstdio>
#include<cstdlib>
#include<map>
#include<set>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath> 
#include<stack>
#include<queue>
#include<iostream>

#define inf 0x3f3f3f3f
using namespace std;
typedef long long LL;
typedef pair<int, int> pr;

int t, c, ts, te;
const int maxn = 2500;
const int maxm = 6205;
struct edge{
    int to, nxt, cost;
}e[maxm * 2];
int head[maxn], tot;

void add(int x, int y, int w)
{
    e[++tot].to = y;
    e[tot].cost = w;
    e[tot].nxt = head[x];
    head[x] = tot;
    e[++tot].to = x;
    e[tot].cost = w;
    e[tot].nxt = head[y];
    head[y] = tot;
}

bool vis[maxn];
int d[maxn]; 
void dijkstra()
{
    memset(d, 0x3f, sizeof(d));
    for(int i = head[ts]; i; i = e[i].nxt){
        d[e[i].to] = e[i].cost;
    }
    vis[ts] = true;
    d[ts] = 0;
    for(int i = 1; i < t; i++){
        int min = inf, min_id;
        for(int j = 1; j <= t; j++){
            if(d[j] < min && !vis[j]){
                min = d[j];
                min_id = j;
            }
        }
        vis[min_id] = true;
        for(int i = head[min_id]; i; i = e[i].nxt){
            if(d[e[i].to] > min + e[i].cost){
                d[e[i].to] = min + e[i].cost;
            }
        }
    }
}

int main()
{
    scanf("%d%d%d%d", &t, &c, &ts, &te);
    for(int i = 0; i < c; i++){
        int rs, re, ci;
        scanf("%d%d%d", &rs, &re, &ci);
        add(rs, re, ci);
    }
    
    dijkstra();
    printf("%d
", d[te]);
}