CCF 201609-4 交通規劃 傳送門

先放張圖得瑟得瑟, 一遍過, so happy.

這是一道描述非常簡單的題目, 看到題目就有點小開心, 覺得肯定能做出來的.

目的很清晰, 所有的點到首都的距離都是最短路, 那麼最短路就要儲存下來捨棄其他無用的邊. 看資料範圍, 就用SPFA演算法了. 中心思想是記錄最短路徑+去除無用邊

關鍵是一條路的最短路可能會有好幾條, 那麼應該怎麼選呢? 如果停留在表面, 會覺得這道題目非常的複雜, 因為選出這些邊, 似乎會影響後面的結果, 到底選哪些邊? 計算機可不知道, 這裡面一定有一個規則.

畫張圖, 仔細觀察後會發現, 一個點可以有幾個出度(也可以有0個), 但是隻會有一個入度. 因為>1的話是無意義的. 那麼很清晰了, 我們選出每個點入邊裡面最短的邊即可, 也就是總共選出n - 1條邊

#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;

struct Node {
    int to, w;
    Node (int t, int w) : to(t), w(w) {}
};

struct Edge {
    int from, to, w;
    Edge (int f, int t, int w) : from(f), to(t), w(w) {}
};

const
 
 int maxn = 10000 + 5;
const int INF = 0x3f3f3f3f;
vector<Node> G[maxn], last[maxn];
vector<Edge> edge;
int inque[maxn] = {}, n, m, dis[maxn], ans = 0;

bool cmp(Edge x, Edge y)
{
    if (x.to != y.to) return x.to < y.to;
    return x.w < y.w;
}

void SPFA()
{
    queue<int> Q;
    Q.push(1
 
);
    inque[1] = true;
    dis[1] = 0;
    while (!Q.empty()) {
        int now = Q.front();
        inque[now] = false;
        Q.pop();
        for (int i = 0; i < G[now].size(); ++i) {
            int v = G[now][i].to, w = G[now][i].w;
            if (dis[v] == dis[now] + w) {
                last[v].push_back(Node(now, w));
            } else if (dis[v] > dis[now] + w) {
                last[v].clear();
                last[v].push_back(Node(now, w));
                dis[v] = dis[now] + w;
                if (!inque[v]) {
                    Q.push(v);
                    inque[v] = true;
                }
            }
        }
    }
}

int main()
{
    memset(dis, 0x3f, sizeof(dis));
    cin >> n >> m;
    for (int i = 1, u, v, w; i <= m; ++i) {
        cin >> u >> v >> w;
        G[u].push_back(Node(v, w));
        G[v].push_back(Node(u, w));
    }
    SPFA();
    for (int i = 1; i <= n; ++i) {
        for (int j = 0; j < last[i].size(); ++j) {
            edge.push_back(Edge(last[i][j].to, i, last[i][j].w));
        }
    }
    sort(edge.begin(), edge.end(), cmp);
    int to = 2;
    for (int i = 0; i < edge.size(); ++i) {
        if (edge[i].to == to) {
            ans += edge[i].w;
            to++;
        }
    }
    cout << ans;
}