【D - Constructing Roads】

思路：

• 最小生成树，已经建好的路 mp 置零即可。

代码：

• 282ms 732kB

​//282ms		732kB


#include <iostream>
#include <cstring>
#include <queue>
#define INF 0x3f3f3f3f

using namespace std;

const int maxn = 105;

int N,M;
int ans;
bool vis[maxn];
int mp[maxn][maxn];
int dis[maxn];
struct NODE{
	int id;
	int dis;
	NODE(int id,int dis) : id(id) , dis(dis) {} ;
	friend bool operator > (const NODE &a , const NODE &b){
		return a.dis > b.dis;
	}
};
priority_queue <NODE , vector<NODE> , greater<NODE> > Q;

void INIT(){
	ans = 0;
	memset(vis , 0 , sizeof(vis));
	memset(dis , INF , sizeof(dis));
	return ;
}

void PRIM(){
	dis[1] = 0;
	Q.push(NODE(1 , 0));
	while(Q.size()){
		NODE cur = Q.top() ; Q.pop() ;
		int id = cur.id;
		if(vis[id])
			continue;
		vis[id] = true;
		ans += dis[id];
		for(int i=1;i<=N;i++)
			if(!vis[i] && dis[i] > mp[id][i]){
				dis[i] = mp[id][i];
				Q.push(NODE(i , dis[i]));
			}
	}
	return ;
}

int main(){
	INIT();
	cin>>N;
	for(int i=1;i<=N;i++)
		for(int j=1;j<=N;j++)
			cin>>mp[i][j];
	cin>>M;
	while(M--){
		int l,r;
		cin>>l>>r;
		mp[l][r] = mp[r][l] = 0;
	}
	PRIM();
	cout<<ans<<endl;
}​