洛谷P3905 道路重建

道路重建

题目描述

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

输入格式

输入文件第一行为一个整数 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，代表需要恢复交通的两个重要城市。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

1

代码： 

1.spfa

#include <bits/stdc++.h>
using namespace std;
#define ing long long
int n,m,d,A,b;
int dis[10010];//存到某个点要修的路的距离 
bool vis[10010];//标记某个点是否在队列里 
bool po[10010][10010];//标记某条路是否被破坏 
vector<pair<int,int> > e[10010];//存路 
void spfa(){
	dis[A]=0;
	vis[A]=1;
	queue<int> q;
	q.push(A);
	while(q.size()){
		int now=q.front();
		q.pop();
		vis[now]=0;
		for(auto t:e[now]){
			int a=t.first,b=0;//a是now能到达的点，注意先将b赋为0 ！！ 
			if(po[now][a]==1) b=t.second;
			//如果从now到a的道路被破坏，这段距离就要修，b才赋值为道路的长 
			if(dis[a]>b+dis[now]){
				dis[a]=b+dis[now];//求较小值 
				if(vis[a]==0){//如果a点不在队列里，入队 
					vis[a]=1;
					q.push(a);
				}
			}
		}
	}
}
signed main(){
	cin >> n >> m;
	for(int i = 1;i<=m;i++){
		int u,v,w;
		cin >> u >> v >> w;
		e[u].push_back({v,w});//建双向边 
		e[v].push_back({u,w});
	}
	cin >> d;
	for(int i = 1;i<=d;i++){
		int u,v;
		cin >> u >> v;
		po[u][v]=1;//标记被破坏的路 
		po[v][u]=1;
	}
	for(int i = 1;i<=n;i++) dis[i]=1e18;//初始化为极大值 
	cin >> A >> b;
	spfa();
	cout << dis[b];
	return 0;
} 

2.dijkstra 

#include <bits/stdc++.h>
#define PII pair<int,int>//宏定义，方便写 
#define fi first
#define se second
using namespace std;
int n,m,d,a,b;
bool po[110][110];
int dis[1000010];
bool vis[1000010];
vector<pair<int,int> > e[1000010];
void distra(){//模板 小改 
	priority_queue<PII,vector<PII>,greater<PII> > q;
	for(int i = 1;i<=n;i++) dis[i]=1e18;
	q.push({0,a});
	dis[a]=0;
	while(q.size()){
		auto t=q.top();
		q.pop();
		int now=t.se;
		if(vis[now]==1) continue;
		vis[now]=1;
		for(auto tt:e[now]){
			int spot=tt.se,w=0;
			if(po[now][spot]==1) w=tt.fi;
			//如果路被破坏了才要加上这段路的长 
			if(dis[spot]>dis[now]+w){
				dis[spot]=dis[now]+w;
				q.push({dis[spot],spot}); 
			}
		}
	}
}
int main(){
	cin >> n >> m;
	for(int i = 1;i<=m;i++){
		int u,v,w;
		cin >> u >> v >> w;
		e[u].push_back({w,v});//道路是双向的 
		e[v].push_back({w,u});
	}
	cin >> d;
	for(int i = 1;i<=d;i++){
		int c,d;
		cin >> c >> d;
		po[c][d]=1;
		po[d][c]=1;
	}
	cin >> a >> b;
	distra();//dijkstra
	cout << dis[b];
	return 0;
}