LG 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

---

跑一遍spfa
如果道路损坏就加上这条路的权值
否则加0

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define For(i,l,r) for(int i=l;i<=r;++i)
using namespace std;
int read()
{
    char c;
    int a=0;
    bool flag=0;
    while((c=getchar())==' '||c=='\r'||c=='\n');
    if(c=='-')
    {
        flag=1;
        c=getchar();
    }
    while(isdigit(c))
    {
        a*=10;
        a+=c;
        a-='0';
        c=getchar();
    }
    return a*(flag?-1:1);
}
struct line{
    int next,to,v;
}edge[10001];
int n,m,last[101],dis[101],que[100010],qh,qt;
bool brk[101][101];
void add(int from,int to,int val,int i)
{
    edge[i].next=last[from];
    last[from]=i;
    edge[i].to=to;
    edge[i].v=val;
}
int main()
{
    int tx,ty,tz,d,a,b;
    n=read();m=read();
    For(i,1,n)
    dis[i]=2140483647;
    For(i,1,m*2)
    {
        cin>>tx>>ty>>tz;
        add(tx,ty,tz,i++);
        add(ty,tx,tz,i);
    }
    cin>>d;
    For(i,1,d)
    {
        cin>>tx>>ty;
        brk[tx][ty]=1;
        brk[ty][tx]=1;
    }
    cin>>a>>b;
    dis[a]=0;
    que[qt++]=a;
    while(qh<qt)
    {
        tx=last[que[qh]];
        while(tx)
        {
            if(dis[edge[tx].to]>dis[que[qh]]+(brk[que[qh]][edge[tx].to]?edge[tx].v:0))
            {
                que[qt++]=edge[tx].to;
                dis[edge[tx].to]=dis[que[qh]]+(brk[que[qh]][edge[tx].to]?edge[tx].v:0);
            }
            tx=edge[tx].next;
        }
        ++qh;
    }
    cout<<dis[b];
    return 0;
}