【CODEVS】1557 - 热浪 （SPFA）

题目链接

其实就是一道单源最短路的题目，就是想测试一下我的SPFA板子。

双向边看成单向WA2。。。

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int ,int> pii;
const int INF = 0x3f3f3f3f;
const int N = 3e3+10;
const int M = 7e3+10;

struct EDGE{
    int next;
    int to;
    int w;
}edge[M];

int head[N],dis[N];
bool inq[N];
int T,C,Ts,Te,cnt = 0;

void add(int u, int v, int w){
    edge[++cnt].next = head[u];
    edge[cnt].to = v;
    edge[cnt].w  = w;
    head[u] = cnt;
}

void SPFA(){
    memset(dis, INF, sizeof dis);
    dis[Ts] = 0;
    int p; queue<int > que; 
    que.push(Ts); inq[Ts] = true;
    while(!que.empty()){
        p = que.front();que.pop();inq[p] = false;
        for(int i = head[p]; ~i; i = edge[i].next){
            int u = edge[i].to;
            if(dis[u] > dis[p] + edge[i].w){
                dis[u] = dis[p] + edge[i].w;
                if(!inq[u]){
                    inq[u] = true;
                    que.push(u);
                }
            }
        }
    }
    printf("%d\n", dis[Te]);
}
int main(int argc, char const *argv[])
{
    memset(head, -1, sizeof head);
    cnt = 0;
    int u, v, w;
    scanf("%d %d %d %d",&T, &C, &Ts, &Te);
    for(int i=1;i<=C;++i){
        scanf("%d %d %d",&u, &v, &w);
        add(u, v, w);
        add(v, u, w);
    }
    SPFA();
    return 0;
}
/*
7 11 5 4
2 4 2
1 4 3
7 2 2
3 4 3
5 7 5
7 3 3
6 1 1
6 3 4
2 4 3
5 6 3
7 2 1

7
*/