Dijkstra算法--变型题的求解

Dijkstra算法–变型题的求解

//Dijkstra算法用来求最短距离 已经实现了//做题题型有变化，需要处理升级版/*//碰到有两条及以上可以达到最短距离的路径，题目就会给出第二标尺 //第一标尺是距离，要求在所有最短路径中选择第二标尺最优的一条路径。////  第二标尺常见的有下面3种出题方法或组合//      1.给每条边再增加一个边权（比如花费），然后要求在最短路径有//  多条时要求路径上的花费之和最小。（或者最大） //      2.给每个点增加一个点权(例如每个城市能收集到的物资)，然后在 //  最短路径有多条时要求路径上的点权之和最大（或最小） //      3.直接问有多少条最短路径。//      对以上3中问题，都只需要增加一个数组来存放新增的边权或点权或//  最短路径条数，然后在Dijkstra算法中修改优化d[v]的那个步骤即可，其他//  部分不需要改动。 */ //1.新增边权//初始时，只有c[s]=0,其余c[u]均为INF for(int v=0;v<n;v++) {    //如果v未访问  && u能到达v    if( vis[v]==false  && G[u][v]!=INF  )     {        //以u为中介点可以使d[v]更优         if( d[u]+G[u][v] < d[v] )        {            d[v]=d[u]+G[u][v];            c[v]=c[u]+cost[u][v];        }        else if( d[u]+G[u][v]==d[v] && c[u]+cost[u][v] <c[v] )        {            //最短距离相同时，看能否使c[v]更优             c[v]=c[u]+cost[u][v];        }    }}//2.新增点权//这里要注意，我们要求物资的最大情况了//用weight[u]表示城市u中的物资数目（由题目输入）//增加一个数组w[],令从起点s达到顶点u可以收集到的最大物资为w[u]//初始时，只有w[s]为weight[s],其余w[u]均为0 for(int v=0;v<n;v++){    //如果v未被访问 && u能到达v    if( vis[v]==false  && G[u][v]!=INF )     {        if(d[u]+G[u][v]<d[v])        {            d[v]=d[u]+G[u][v];            w[v]=w[u]+weight[v];        }        else if( d[u] + G[u][v] == d[v]  && w[u]+weight[v]>w[v] )        {            //最短距离相同时，看能否使w[v]更优            w[v]=w[u]+weight[v];        }    }}//3.求最短路径条数//需要增加一个数组num[]，令从起点s到达顶点u的最短路径条数为num[u]//初始化时，只有num[s]=1,其余num[u]均为0。//最后，要注意，计算路径数的时候要用 替换 或者 累加for(int v=0;v<n;v++) {    if( vis[v]==false && G[u][v]!=INF  )    {        if(d[u]+G[u][v] <d[v]  )        {            d[v]=d[u]+G[u][v];            num[v]=num[u];        }        else if( d[u]+G[u][v]==d[v] )        {            //最短距离相同时累加num             num[v]+=num[u];        }    }}