CCF CSP——交通规划(201809-4) & 复习Dijkstra的两种实现

问题描述
  G国国王来中国参观后,被中国的高速铁路深深的震撼,决定为自己的国家也建设一个高速铁路系统。
  建设高速铁路投入非常大,为了节约建设成本,G国国王决定不新建铁路,而是将已有的铁路改造成高速铁路。现在,请你为G国国王提供一个方案,将现有的一部分铁路改造成高速铁路,使得任何两个城市间都可以通过高速铁路到达,而且从所有城市乘坐高速铁路到首都的最短路程和原来一样长。请你告诉G国国王在这些条件下最少要改造多长的铁路。
输入格式
  输入的第一行包含两个整数n, m,分别表示G国城市的数量和城市间铁路的数量。所有的城市由1到n编号,首都为1号。
  接下来m行,每行三个整数a, b, c,表示城市a和城市b之间有一条长度为c的双向铁路。这条铁路不会经过a和b以外的城市。
输出格式
  输出一行,表示在满足条件的情况下最少要改造的铁路长度。
样例输入

4 5
1 2 4
1 3 5
2 3 2
2 4 3
3 4 2

样例输出

11

评测用例规模与约定
  对于20%的评测用例,1 ≤ n ≤ 10,1 ≤ m ≤ 50;
  对于50%的评测用例,1 ≤ n ≤ 100,1 ≤ m ≤ 5000;
  对于80%的评测用例,1 ≤ n ≤ 1000,1 ≤ m ≤ 50000;
  对于100%的评测用例,1 ≤ n ≤ 10000,1 ≤ m ≤ 100000,1 ≤ a, b ≤ n,1 ≤ c ≤ 1000。输入保证每个城市都可以通过铁路达到首都。

思路
增加一个cost[]数组
cost[i]用来记录最短路径情况下,把点 i 加入到集合S中的最短路径(最短的边长),也即从已处理的点到点 i 的最短距离。

Dijkstra算法中,集合S的含义
集合S:已经确定的顶点
集合T:尚未确定的顶点

我的代码:

#include <iostream>
#include <algorithm>//用于函数min
#include <vector> //用于图的邻接表存储
#include <queue> //用于优先队列
#include <climits>  //用于定义无穷大

using namespace std;

const int MAXN = 10000 + 10;//顶点数最多10000个(加上10个安全量)
const int INF = INT_MAX;  //定义无穷大为最大的整数

struct Edge{  //用于图邻接表存储的边的信息
    int to;  //该边的另外一端的顶点号
    int length; //该边的长度
    Edge(int t,int l):to(t),length(l){}//构造函数
};

struct Node{  //各节点的信息
    int number;//该节点的编号
    int distance;//源点到该节点的最短路径长度
    Node(){}
    Node(int n,int d):number(n),distance(d){}
    //由与点的信息要加入到优先队列中,故应设置各点大小关系
    bool operator<(const Node& point) const{//小于号重载
        return distance > point.distance;//距离大的优先级小
    }
};

vector<Edge> graph[MAXN];//图的邻接表存储
int cost[MAXN];//最短路径前提下,从已处理的点到该点的最短距离
int dist[MAXN];//从源点(首都1)到各点的最短路径
bool visit[MAXN];//标记各个点是否被访问过

void Dijkstra(int start,int n){//Dijkstra算法
    fill(dist,dist+n+1,INF);//将所有最短路径初始化为无穷大
    fill(visit,visit+n+1,false);//将所有点标记为未访问过
    fill(cost,cost+n+1,INF);//将所有点到已有集合的最短距离初始化为INF
    dist[start] = 0;//修改源点的信息
    cost[start] = 0;
    priority_queue<Node>que;
    que.push(Node(1,0));//将源点(首都1)加入到优先队列中
    Node front;
    while(!que.empty()){
        front = que.top();
        que.pop();
        int current = front.number;//current记录对头结点的编号
        if(!visit[current]){//若current没有被访问过
            visit[current] = true;//将其标记为已访问过
            //进行松弛操作,更新与该点相邻的所有点的信息
            for(int i=0;i<graph[current].size();i++){
                //获取current的邻接点now
                int now = graph[current][i].to;
                if(visit[now]){//若该点已被访问过,则处理下一个邻接点
                    continue;
                }
                //记录下current到now的距离
                int spend = graph[current][i].length;
                if(dist[now] > spend + dist[current]){
                    dist[now] = spend + dist[current];
                    cost[now] = spend;
                    //若该点的信息被更新,则加入到优先队列中
                    //一个点可能被添加多次,但通过visit数组
                    //可以保证其只被访问过一次(距离最短的那次)
                    que.push(Node(now,dist[now]));
                }else if(dist[now] == spend + dist[current]){
                    cost[now] = min(cost[now],spend);//若相等,取小
                }
            }
        }
    }
}

int main(){
    int n,m,a,b,c;
    cin>>n>>m;
    for(int i=0;i<m;i++){
        cin>>a>>b>>c;
        graph[a].push_back(Edge(b,c));//无向图
        graph[b].push_back(Edge(a,c));
    }
    Dijkstra(1,n);//传入首都的位置 城市的个数
    int answer = 0;
    for(int i=1;i<=n;i++){//cost[]中各值的和即为答案
        answer += cost[i];
    }
    cout<<answer;
    return 0;
}

知识点复习:
Dijkstra算法的两种实现方式(求点start到点terminal的最短距离)。
优先队列实现

#include <iostream>
#include <queue>
#include <climits>
#include <vector>
#include <algorithm>

using namespace std;

const int MAXN = 10000;
const int INF = INT_MAX;

struct Edge{
    int to;
    int length;
    Edge(int t,int l):to(t),length(l){}
};

struct Node{
    int number;
    int distance;
    Node(){}
    Node(int n,int d):number(n),distance(d){}
    bool operator<(const Node& point) const{
        return distance > point.distance;
    }
};

bool visit[MAXN];
int dist[MAXN];
vector<Edge> graph[MAXN];

void Dijkstra(int start,int n){
    fill(dist,dist+n,INF);
    dist[start] = 0;
    priority_queue<Node>que;
    que.push(Node(start,dist[start]));
    while(!que.empty()){
        int current = que.top().number;
        que.pop();
        for(int i=0;i<graph[current].size();i++){
            int target = graph[current][i].to;
            int distance = graph[current][i].length;
            if(dist[target] < dist[current] + distance){
                dist[target] = dist[current] + distance;
                que.push(Node(target,dist[target]));
            }
        }
    }
}

int main(){
    int n,m,from,to,length,start,terminal;
    cin>>n>>m;
    for(int i=0;i<m;i++){
        cin>>from>>to>>length;
        graph[from].push_back(Edge(to,length));
        graph[to].push_back(Edge(from,length));
    }
    cin>>start>>terminal;
    Dijkstra(start,n);
    if(dist[terminal]==INF){
        cout<<"-1";
    }else{
        cout<<dist[terminal];
    }
    return 0;
}

普通实现

#include <iostream>
#include <queue>
#include <climits>
#include <vector>
#include <algorithm>

using namespace std;

const int MAXN = 10000;
const int INF = INT_MAX;

struct Edge{
    int to;
    int length;
    Edge(int t,int l):to(t),length(l){}
};

bool visit[MAXN];
int dist[MAXN];
vector<Edge> graph[MAXN];

void Dijkstra(int start,int n){
    fill(dist,dist+n,INF);
    fill(visit,visit+n,false);
    dist[start] = 0;
    for(int i=0;i<n;i++){
        int target = -1;
        for(int i=0;i<n;i++){
            if(visit[target]){
                continue;
            }
            if(target==-1 || dist[i]<dist[target]){
                target = i;
            }
        }
        for(int j=0;i<graph[target].size();j++){
            int current = graph[target][j].to;
            int distance = graph[target][i].length;
            if(dist[target] > dist[current] + distance){
                dist[target] = dist[current] + distance;
            }
            visit[target] = true;
        }
    }

}

int main(){
    int n,m,from,to,length,start,terminal;
    cin>>n>>m;
    for(int i=0;i<m;i++){
        cin>>from>>to>>length;
        graph[from].push_back(Edge(to,length));
        graph[to].push_back(Edge(from,length));
    }
    cin>>start>>terminal;
    Dijkstra(start,n);
    if(dist[terminal]==INF){
        cout<<"-1";
    }else{
        cout<<dist[terminal];
    }
    return 0;
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值