洛谷P1119 灾后重建

P1119 灾后重建

题目链接-P1119 灾后重建


解题思路
$floyd思想$

• $folyd$算法没什么说的，就是动态规划，通过枚举中继点来更新最短路
• 因为刚开始每个村庄之间并不是联通的，随着重建村庄才连接起两村庄间的道路，且每次都是询问第t天，从村庄x到村庄y的最短路径长度，所以我们可以边更新边输出
• 在开始需将每个村庄之间的距离都初始化为INF，以边后面最短路的更新及输出-1情况的判断
• 对于输出-1的情况，如果在第$t$天，$x$或$y$村庄还没有重建，则说明两村庄不可能联通，即输出-1，若两村庄都已重建，就判断更新前t天路径后两村庄之间距离是否为INF即可
• 具体操作见代码

附上代码

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e4+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
int n,m,q;
int a[210],e[210][210];
void updata(int k){//以k为中继点，更新最短路
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			if(e[i][j]>e[i][k]+e[k][j])
				e[i][j]=e[i][k]+e[k][j];
		}
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);

	cin>>n>>m;
	for(int i=0;i<n;i++)
		cin>>a[i];
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			e[i][j]=INF;//初始化
	for(int i=0;i<m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		e[u][v]=e[v][u]=w;
	} 
	cin>>q;
	int tmp=0;
	while(q--){
		int x,y,t;
		cin>>x>>y>>t;
		if(a[x]>t||a[y]>t) cout<<-1<<endl;
		else{
			while(a[tmp]<=t&&tmp<n){
			//如果第tmp个村庄重建时间小于等于t且不大于n，就说明可以更新以该村庄为中继点的村庄
				updata(tmp);
				tmp++;//以便判断下一个村庄
			}
			e[x][y]==INF?cout<<-1<<endl:cout<<e[x][y]<<endl;
		}
	}
	return 0;
}