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;
}