1001.Total Eclipse
题意:给出各个城市亮度和连通情况,可以对连在一起的城市的亮度减一或者对一个城市亮度减一,求使所有城市亮度为0的最少操作数。
思路:假设每个城市都是孤立的,这时操作数就是亮度总和。将城市按亮度大小排序,遍历每个城市的相邻城市,如果相邻城市已经遍历过且不在连通块上,操作数减去其亮度同时将其加入连通块。
知识点:并查集
#include<cstdio>
#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn = 4e6+7;
struct node {
int value;
int index;
} k[maxn];
int T,n,m;
vector<int> g[maxn];
bool vis[maxn];
int root[maxn];
int find(int x) {
return x == root[x] ? x : (root[x] = find(root[x]));
}
void merge(int i,int j) {
root[find(i)] = find(j);
}
bool cmp(node i,node j) {
return i.value > j.value;
}
int main() {
scanf("%d",&T);
while(T --) {
long long sum = 0;
scanf("%d%d",&n,&m);
for(int i = 1; i <= n; i ++) {
scanf("%d",&k[i].value);
k[i].index = i;
g[i].clear();
vis[i] = false;
root[i] = i;
sum += k[i].value;
}
sort(k + 1 ,k + n + 1 ,cmp);
for(int i = 0,x,y; i < m; i ++) {
scanf("%d%d",&x,&y);
g[x].push_back(y);
g[y].push_back(x);
}
for(int i = 1; i <= n; i ++) {
for(int j = 0; j < g[k[i].index].size() ; j ++) {
int x = g[k[i].index][j];
if(vis[x] && find(x) != find(k[i].index)) {
merge(x,k[i].index);
sum -= k[i].value;
}
}
vis[k[i].index] = true;
}
printf("%lld\n",sum);
}
return 0;
}