CCF习题 201612-4 交通规划 （Dijkstra + 贪心）

大体题意：

题意不说了 中文的= =

思路：

既然要求到1的最短距离要存在，我们先用dijkstra算法求出每一个点到1位置的最短距离，然后在看要哪一些边！

既然要求路最短 并且包括最短距离，那么我们可以枚举每一条边{u,v,w}当这条路就是v到1位置的最短路之一时，我们就把这条边的权值和已经保存的最小权值进行比较！

这样的贪心方法想一想 是很合理的！

我们贪心的要 每一个点最短路上的最小值，那么最后的答案肯定是合理的，在联通方面，既然每一个点都要找出一个边，那么你无论怎么走，他都是联通的！

详细见代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
const int maxn = 10000 + 10;
const int inf = 0x3f3f3f3f;
struct edge{
    int u, v, w;
    edge(int u = 0,int v = 0,int w = 0):u(u),v(v),w(w){}
};
struct Node{
    int d, u;
    Node(int d = 0, int u = 0):d(d),u(u){}
    bool operator < (const Node & rhs) const {
        return d > rhs.d;
    }
};
priority_queue<Node> q;
int ans[maxn];
struct Dijkstra{
    int n, m;
    bool vis[maxn];
    vector<edge> edges;
    vector<int>G[maxn];
    int d[maxn];
    void init(int n){
        edges.clear();
        for (int i = 0; i < maxn; ++i)G[i].clear();
        this->n = n;
        m = 0;
        memset(vis,0,sizeof vis);
    }
    void addegde(int u,int v,int w){
        edges.push_back(edge(u,v,w));
        G[u].push_back(m++);
    }
    void Go(int s){
        memset(d,inf,sizeof d);
        d[s] = 0;
        while(!q.empty())q.pop();
        q.push(Node(0,s));
        while(!q.empty()){
            Node x = q.top(); q.pop();
            int u = x.u;
            if (vis[u])continue;
            vis[u] = 1;
            for (int i = 0; i < G[u].size(); ++i){
                edge& e = edges[G[u][i]];
                if (d[e.v] > d[e.u] + e.w){
                    d[e.v] = d[e.u] + e.w;
                    q.push(Node(d[e.v],e.v));
                }
            }
        }
    }
}dij;
int main(){
    int n,m;
    scanf("%d %d",&n, &m);
    dij.init(n);
    memset(ans,inf,sizeof ans);
    for (int i = 0; i < m; ++i){
        int u, v, w;
        scanf("%d %d %d",&u, &v, &w);
        dij.addegde(u,v,w);
        dij.addegde(v,u,w);
    }
    dij.Go(1);
    for (int i = 1; i <= n; ++i){
        for (int j = 0; j < dij.G[i].size(); ++j){
            int u,v,w;
            u = dij.edges[dij.G[i][j]].u;
            v = dij.edges[dij.G[i][j]].v;
            w = dij.edges[dij.G[i][j]].w;
            if (dij.d[u] + w == dij.d[v]){
                ans[v] = min(ans[v],w);
            }
        }
    }
//    printf("^^^ %d\n",dij.d[4]);
    int sum = 0;
    for (int i = 2; i <= n; ++i) sum += ans[i];
    printf("%d\n",sum);
    return 0;
}

问题描述

答题栏