文章标题 CSU 1817 ： Bones’s Battery（Floyd+二分）

1817: Bones’s Battery

1817: Bones’s Battery
题意：有n个点m每条边的无向带权图，从点ui到vi需要花费电量di,要求对任一起点s到终点e，都能通过充电并且充电的次数不超过k次(在任何一个点都可以充满电)互相到达，问电池的容量最小是多少。
分析：这是我遇到的第一道二分答案，所以一开始不知道怎么做，看了标程后恍然大悟。。
首先，已经明确二分答案了，那么关键就是要怎么判断但前值的可行性
先用Floyd跑一遍最短路，将任意两点的最短距离求出来
然后二分答案，然后对于当前二分的值mid,我们通过再开一个数组red[][],然后将red[i][i]赋为0，然后对于任意两点u–>v,如果当前的路径长度小于x则赋值为1，否则赋值为INF。然后再跑一遍red的最短路，然后求出red数组中的最大值max，如果max<=k时，说明当前x还可以变小，所以往小的区间二分，否则说明当前的x太小了不能满足任意两个点通过充k次点来到达，所以得往大的区间二分。
最后将答案二分出来就行了。
代码：

#include<iostream>
#include<string>
#include<cstdio>
#include<cstring>
#include<vector>
#include<math.h>
#include<map>
#include<queue> 
#include<algorithm>

using namespace std;

const double pi=acos(-1.0);
const int inf = 0x3f3f3f3f;
const int maxn=100005;
const double eps=1e-8; 
const long long INF=10000000000;

int n,k,m;
long long dis[105][105], red[105][105]; 

void init(long long dis[][105]){//初始化数组 
    for (int i=0;i<n;i++){
        for (int j=0;j<n;j++){
            if (i==j)dis[i][j]=0;
            else dis[i][j]=INF;
        }
    }
}

void floyd(long long dis[][105]){//folyd 求各源点间的最短距离 
    for(int t=0;t<n;t++){
        for (int i=0;i<n;i++){
            for (int j=0;j<n;j++){
                dis[i][j]=min(dis[i][j],dis[i][t]+dis[t][j]);
            }
        }
    }
}

int main()
{
    int t;
    scanf ("%d",&t);
    while (t--){
        scanf ("%d%d%d",&n,&k,&m);
        for (int i=0;i<n;i++){
            for (int j=0;j<n;j++){
                if (i==j)dis[i][j]=0;
                else dis[i][j]=INF;
            }
        }
        int u,v,val;
        for (int i=0;i<m;i++){
            scanf ("%d%d%d",&u,&v,&val);
            dis[u][v]=dis[v][u]=val;
        }
        floyd(dis);
        long long lo=1,hi=INF;
        long long mid;
        long long ans=INF;
        while (lo<=hi){//二分答案 
            mid=(lo+hi)/2;
            init(red);
            for (int i=0;i<n;i++){
                for (int j=0;j<n;j++){
                    red[i][j]= dis[i][j]<=mid?1:INF;//如果小于mid说明i->j只需要充一次电 
                }
            }
            floyd(red);//跑floyd 
            long long d=0;
            for (int i=0;i<n;i++){
                for (int j=0;j<n;j++){
                    if (red[i][j]>d)d=red[i][j];//求出充电次数最多的一条路径 
                }
            }
            if (d<=k){//如果次数小于k说明此时的距离符合 
                ans=mid;//保存 
                hi=mid-1;
            }
            else lo=mid+1;
        }
        printf ("%lld\n",ans);//输出 
    }
    return 0;
 }