CSP 交通规划

#include <iostream>
#include<cstring>
#include<queue>

using namespace std;

// 想到了方法spfa，但是感觉不知道怎么构造之后那个判断条件，以及不知道这种论证是否成立。


const int N = 2e5 + 10,INF = 0x3f3f3f3f; // 无向图，两倍

int n;      // 总点数
int h[N], w[N], e[N], ne[N], idx;       // 邻接表存储所有边
int dist[N];        // 存储每个点到1号点的最短距离
bool st[N];     // 存储每个点是否在队列中

int m;

void add(int a,int b,int c)
{
    e[idx] = b,w[idx] = c,ne[idx] = h[a] , h[a] = idx ++;
}

// 求1号点到n号点的最短路距离，如果从1号点无法走到n号点则返回-1
void spfa()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;

    queue<int> q;
    q.push(1);
    st[1] = true;

    while (q.size())
    {
        auto t = q.front();
        q.pop();

        st[t] = false;

        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (dist[j] > dist[t] + w[i])
            {
                dist[j] = dist[t] + w[i];
                if (!st[j])     // 如果队列中已存在j，则不需要将j重复插入
                {
                    q.push(j);
                    st[j] = true;
                }
            }
        }
    }
}


int main()
{
    cin >> n >> m;
    memset(h,-1,sizeof h);
    while(m--)
    {
        int a,b,c;
        cin >> a >> b >> c;
        add(a,b,c);
        add(b,a,c);
    }
    spfa();
    int ans = 0;
    for(int k = 2;k <= n;k++)
    {
        int roads = INF;
        for (int i = h[k]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(dist[k] == dist[j] + w[i])
            {
                roads = min(roads,w[i]);
            }
        }
        ans += roads;
    }
    cout << ans;

    return 0;
}