P3905 道路重建（洛谷）

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

输入格式
输入文件第一行为一个整数n  (2<n≤100)，表示城市的个数。这些城市编号从 1 到 n。

第二行为一个整数 
m (n−1≤m≤ (1/2)*n(n−1))，表示道路的数目。

接下来的 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，代表需要恢复交通的两个重要城市。

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

输入输出样例
输入 #1复制
3
2
1 2 1
2 3 2
1
1 2
1 3
输出 #1复制
1

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
 
using namespace std;
 
const int N = 1e6+10;
int A,B;
int n,m,d;
int dis[N];
int vis[N]; 
int cnt[N];
int a,b,c;
map<pair<int,int> ,int>mp;
vector<pair<int,int >> e[N];
void spfa()
{
	for(int i=1;i<=n;i++) dis[i]=1e18;
	queue<int>q;
	q.push(A);
	dis[A] = 0;
	vis[A] = 1;
	while(q.size())
	{
		int now = q.front();
		q.pop();
		vis[now]=0;
		for(auto t:e[now])
		{
			int spot = t.first,w = 0;
			if(mp[{now,spot}]) w=t.second;
			if(dis[spot]>dis[now]+w)
			{
				dis[spot] = dis[now]+w;
				if(vis[spot] == 0)
				{
					vis[spot] = 1;
					q.push(spot);
				} 
			}
		}
	}
}
 
signed main()
{
	IOS;

	cin>>n>>m;
	while(m--)
	{
		cin>>a>>b>>c;
		e[a].push_back({b,c});
		e[b].push_back({a,c});
		
	}
	cin>>d;
	while(d--)
	{
	
		cin>>a>>b;
		mp[{a,b}]=1;
		mp[{b,a}]=1;
	}
	
	cin>>A>>B;
	spfa();
	cout<<dis[B];
	return 0;
}