当前位置: 代码迷 >> 综合 >> P1119 灾后重建 最短路 floyd算法
  详细解决方案

P1119 灾后重建 最短路 floyd算法

热度:62   发布时间:2023-12-06 11:20:10.0

题目:
在这里插入图片描述
在这里插入图片描述
思路:
一看这道题就是个多源最短路问题,考虑采用floyd算法
如果每次都用一次floyd,显然大数据会tle
所以先一遍把所有的情况都算出来,后面再去调用
如果day不是递增的数组,需要先排序
AC代码:

#include<cstdio>
#include <iostream>
#include<cstring>
#include<algorithm>
#include <queue>
#include <vector>
#include <cmath>
using namespace std;
const int maxn=211;
int n,m,que,day[maxn],f[maxn][maxn][maxn];
void floyd(int x,int y,int t)
{
    for(int k=1;k<=n;k++){
    for(int i=1;i<=n;i++){
    for(int j=1;j<=n;j++){
    if(day[k]<=t){
    f[k][i][j]=min(f[k-1][i][j],f[k-1][i][k]+f[k-1][k][j]);//cout<<"k="<<k<<" "<<"i="<<i<<" "<<"j="<<j<<" "<<"dis[k][i][j]="<<dis[k][i][j]<<endl;}}}}
}
int main()
{
    cin>>n>>m;for(int i=1;i<=n;i++) cin>>day[i];memset(f,0x3f,sizeof(f));for(int i=1;i<=m;i++){
    int u,v,w;cin>>u>>v>>w;u++;v++;f[0][u][v]=f[0][v][u]=w;}for(int i=1;i<=n;i++)for(int j=1;j<=n;j++) f[i][j][j]=0;floyd(n,n,day[n]);cin>>que;for(int i=1;i<=que;i++){
    int x,y,t;cin>>x>>y>>t;x++;y++;if(day[x]>t||day[y]>t) cout<<-1<<endl;else {
    int temp=0;for(int j=1;j<=n;j++) if(day[j]<=t) temp=j;if(f[temp][x][y]==0x3f3f3f3f) cout<<-1<<endl;else cout<<f[temp][x][y]<<endl;}}return 0;
}

复习版:
复习的过程中,还是发现了不少容易错误的点,首先就是由于每次都进行floyd会导致tle。所以其实可以一次把所有情况都算出来,但因为t有限制,所以f必须保留成三维数组不能精简成二维。同时,考虑f[k][i][j]的含义是,i-j的路径里,经过1-k的点,所以i,j的数组都得从1,1开始存,而f[0][i][j]则为i-j的初始路径长度,还得注意,当i=j时,f[k][i][j]=0

#include <bits/stdc++.h>
using namespace std;
const int maxn=211;
int n,m,a[maxn],f[maxn][maxn][maxn],q;
void floyd(int x,int y)
{
    for(int k=1;k<=n;k++){
    for(int i=1;i<=n;i++){
    for(int j=1;j<=n;j++){
    f[k][i][j]=min(f[k-1][i][j],f[k-1][i][k]+f[k-1][k][j]);//cout<<"i="<<i<<" "<<"j="<<j<<" "<<f[i][j]<<endl;}}}}
int main()
{
    cin>>n>>m;for(int i=1;i<=n;i++) cin>>a[i];memset(f,0x3f3f3f3f,sizeof(f));for(int i=1;i<=m;i++){
    int x,y,z;cin>>x>>y>>z;x++;y++;f[0][x][y]=f[0][y][x]=z;}for(int i=1;i<=n;i++){
    for(int j=1;j<=n;j++) f[i][j][j]=0;}floyd(n,n);cin>>q;for(int i=1;i<=q;i++){
    int x,y,t;cin>>x>>y>>t;x++;y++;if(a[x]>t||a[y]>t) {
    cout<<-1<<endl;}else{
    int temp=0;for(int j=1;j<=n;j++) if(a[j]<=t) temp=j;if(f[temp][x][y]==0x3f3f3f3f) cout<<-1<<endl;else cout<<f[temp][x][y]<<endl;}}return 0;
}