练习39，P1119 灾后重建【Floyd算法/二分】

P1119 灾后重建 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

因为每个村庄重建完成的时间和每次询问的时间都是不下降的，所以按照每次询问的时间更新Floyd就行

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define NOTLE ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define endl '\n'
#define T int TT;cin >> TT;while (TT--)
#define lint long long
#define pb push_back
#define eb emplace_back
int t[210],mp[210][210];
int main(){
    NOTLE;
    memset(mp,INF,sizeof(mp)); //初始都设为最大
    int n,m,q;
    cin >> n >> m;
    for(int i=1;i<=n;i++) cin >> t[i];
    for(int i=1;i<=m;i++){
    	int a,b,c;
    	cin >> a >> b >> c;
    	mp[a+1][b+1]=mp[b+1][a+1]=c; //无向图正反都存
    }
    cin >> q;
    int la=0;//la记录上一次更新到第几个点
    while(q--){
    	int x,y,z;
    	cin >> x >> y >> z;
    	
    	int l=la,r=n; //二分搜出该次询问的时间最多能满足多少个点
    	while(l<r){
    		int mid=l+r+1>>1;
    		if(t[mid]<=z) l=mid;
    		else r=mid-1;
    	}
    	
		for(int i=la+1;i<=r;i++) //Floyd更新
			for(int j=1;j<=n;j++)
				for(int k=1;k<=n;k++)
					if(mp[j][k]>mp[j][i]+mp[i][k]) mp[j][k]=mp[j][i]+mp[i][k];
		la=r;//记录这次更新到哪
		if(t[x+1]>z || t[y+1]>z || mp[x+1][y+1]==INF) cout << -1 << endl;
		else cout << mp[x+1][y+1] << endl;
    }
    return 0;
}