Pregunta: Dados n puntos, los pesos son solo 1 y 2, que no están conectados entre sí. Elija 3 puntos de ellos, asegúrese de que la suma de los pesos de los 3 puntos no sea menor que 5, y que los 3 puntos no estén conectados entre sí, y calcule el número de opciones diferentes. Luego agregue n − 1 aristas, conecte dos bloques conectados cada vez y luego genere el número de soluciones diferentes en el estado actual.
Solución del problema: conjunto de búsqueda de unión,
reste la contribución de los dos bloques conectados fusionados cada vez.
En el entrenamiento de la tarde, porque saqué el wa restante antes de la división, muchos tiros ...
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int Parent[maxn], Rank[maxn], cnt1[maxn], cnt2[maxn];
int Find(int x) {
return Parent[x] == x ? x : Parent[x] = Find(Parent[x]);
}
int Union(int x, int y) {
int u, v, root;
u = Find(x);
v = Find(y);
if (u == v) return 0;
if (Rank[u] <= Rank[v]) {
cnt1[v] += cnt1[u];
cnt2[v] += cnt2[u];
root = Parent[u] = v;
if (Rank[u] == Rank[v]) Rank[v]++;
}
else {
cnt1[u] += cnt1[v];
cnt2[u] += cnt2[v];
root = Parent[v] = u;
}
return root;
}
int t, n, a[maxn], u, v;
int main() {
scanf("%d", &t);
while (t--) {
memset(cnt1, 0, sizeof(cnt1));
memset(cnt2, 0, sizeof(cnt2));
memset(Rank, 0, sizeof(Rank));
scanf("%d", &n);
int num1 = 0, num2 = 0;
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
if (a[i] == 1) cnt1[i] = 1, ++num1;
else cnt2[i] = 1, ++num2;
Parent[i] = i;
}
int ini = (1ll * num2 * (num2 - 1) / 2 % mod * num1 % mod +
1ll * num2 * (num2 - 1) * (num2 - 2) / 6 % mod) % mod;
printf("%d\n", ini);
for (int i = 1; i < n; i++) {
scanf("%d%d", &u, &v);
int fau = Find(u), fav = Find(v);
int yu1 = cnt1[fau];
int yu2 = cnt2[fau];
int yv1 = cnt1[fav];
int yv2 = cnt2[fav];
int rt = Union(u, v);
if (i >= n - 2) {
puts("0");
continue;
}
int n1 = cnt1[rt];
int n2 = cnt2[rt];
ini -= 1ll * (n - n1 - n2) * yu2 % mod * yv2 % mod; if (ini < 0) ini += mod;
ini -= 1ll * yu1 * yv2 % mod * (num2 - n2) % mod; if (ini < 0) ini += mod;
ini -= 1ll * yv1 * yu2 % mod * (num2 - n2) % mod; if (ini < 0) ini += mod;
printf("%d\n", ini);
}
}
return 0;
}