KY107最短路径

一、题目描述

描述
N个城市，标号从0到N-1，M条道路，第K条道路（K从0开始）的长度为2^K，求编号为0的城市到其他城市的最短距离
输入描述：
第一行两个正整数N（2<=N<=100）M(M<=500),表示有N个城市，M条道路 接下来M行两个整数，表示相连的两个城市的编号
输出描述：
N-1行，表示0号城市到其他城市的最短路，如果无法到达，输出-1，数值太大的以MOD 100000 的结果输出。
示例1
输入：

4 4
1 2
2 3
1 3
0 1

输出：

8
9
11

二、思路和代码

思路：因为题中提到每次加入的路径长度为2的k次方，故先出现的所有路径和都小于新加入的路径。总体思路与Prim相似。维护一个已知距0点的最短路径的端点的集合，每次都遍历边找到将一端p1在集合内，一端p2在集合外的最短边，更新最短路径，并将p2加入集合。

#include <iostream>
#include <vector>
#include <map>
using namespace std;
//思路：先出现的路径一定先用，因为pow(2,n)>pow(2, n-1)-1=pow(2,0)+pow(2,1)……pow(2, n-1)
//大致思路还是用dijsktra，但每次加入最小边中的其中一个端点必须在已移入的集合中

typedef struct Edge{
    //两个端点
    int p1;
    int p2;
    //2的指数
    int value;
    //pow(2, value)%100000
    int truev;
}Edge;


int main(){
    int N, M, x, y, count;
    map<int, bool> done;
    vector<Edge> edges;
    cin>>N>>M;
    //初始化result，result[i]存0-i的距离
    int result[N];
    for(int i=0; i<N;i++){
        result[i]=-1;
    }
    result[0]=0;
    //初始化edges，存所有边
    edges.resize(M);
    count = 1;
    for(int i=0;i<M;i++){
        cin>>x>>y;
        edges[i].p1=x;
        edges[i].p2=y;
        edges[i].value=i;
        edges[i].truev = count;
        count = count*2%100000;
    }
    //初始化done，done[i]值表示该点是否已移入
    for(int i=0;i<N;i++){
        done[i] = false;
    }
    done[0]=true;
    //循环遍历边，每次加入一条边
    int i;
    int shift;
    while(shift!=-1){
        //i为下标，shift记录要移入的边下标
        i = 0;
        shift = -1;
        //遍历边,找到移入点的下标shift
        while(i<edges.size()){
            //两个顶点都在已移入集合中,不考虑它
            if(done[edges[i].p1] && done[edges[i].p2]){
                i++;
                continue;
            }
            //有一个顶点在已移入集合中，可考虑其是否替换shift
            if(done[edges[i].p1] || done[edges[i].p2]){
                //通过指数判断路径长短，若路径更短则替换
                if(shift==-1 || edges[shift].value>edges[i].value){
                    shift = i;
                }
            }
            i++;
        }
        //可以移入
        if(shift!=-1){
            //如果p1是已经移入的点
            if(done[edges[shift].p1]){
                //更新p2的result和done
                result[edges[shift].p2]=(result[edges[shift].p1]+edges[shift].truev)%100000;
                done[edges[shift].p2]=true;
            }
            //p2是已经移入的点
            else{
                //更新p1的result和done
                result[edges[shift].p1]=(result[edges[shift].p2]+edges[shift].truev)%100000;
                done[edges[shift].p1]=true;
            }
        }
    }
    //输出
    for(int i=1;i<N;i++){
        cout<<result[i]<<endl;
    }
}