洛谷P3905 道路重建

题目描述

从前，在一个王国中，在n个城市间有m条道路连接，而且任意两个城市之间至多有一条道路直接相连。在经过一次严重的战争之后，有d条道路被破坏了。国王想要修复国家的道路系统，现在有两个重要城市A和B之间的交通中断，国王希望尽快的恢复两个城市之间的连接。你的任务就是修复一些道路使A与B之间的连接恢复，并要求修复的道路长度最小。

输入输出格式

输入格式：

输入文件road.in，第一行为一个整数n（2<n≤100）,表示城市的个数。这些城市编号从1到n。第二行为一个整数m（n-1≤m≤n(n-1)/2），表示道路的数目。接下来的m行，每行3个整数i，j，k（1≤i，j≤n，i≠j，0<k≤100），表示城市i与j之间有一条长为k的道路相连。

接下来一行为一个整数d（1≤d≤m），表示战后被破坏的道路的数目。在接下来的d行中，每行两个整数i和j，表示城市i与j之间直接相连的道路被破坏。

最后一行为两个整数A和B，代表需要恢复交通的两个重要城市。

输出格式：

输出文件road.out，仅一个整数，表示恢复A与B间的交通需要修复的道路总长度的最小值。

输入输出样例

输入样例#1： 

3
2
1 2 1
2 3 2
1
1 2
1 3

输出样例#1：  

1

典型的最短路。。。

未被摧毁的路设为0，被摧毁的路不变，再跑 Floyd。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAX 999999999
#define MAXN 110
using namespace std;
int n,m,d,s,t,a[MAXN][MAXN];
bool b[MAXN][MAXN];
inline int read(){
       int date=0,w=1;char c=0;
       while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
       while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
       return date*w;
}
void floyd(){
     for(int k=1;k<=n;k++)
     for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
     a[i][j]=min(a[i][j],a[i][k]+a[k][j]);
     printf("%d\n",a[s][t]);
}
int main(){
    int u,v,w;
    n=read();m=read();
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    a[i][j]=(i==j)?0:MAX;
    for(int i=1;i<=m;i++){
            u=read();v=read();w=read();
            if(a[u][v]>w)a[u][v]=a[v][u]=w;
            }
    d=read();
    for(int i=1;i<=d;i++){
            u=read();v=read();
            b[u][v]=b[v][u]=true;
            }
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    if(!b[i][j]&&a[i][j]!=0&&a[i][j]!=MAX)a[i][j]=0;
    s=read();t=read();
    floyd();
    return 0;
}