Gym 102012A Rikka with Minimum Spanning Trees 解题报告

Gym 102012A Rikka with Minimum Spanning Trees 解题报告

解题思路：读题就读了好久，看吐了，题意是用给你的代码随机生成无向图(m条连接两个节点的边)，求树节点数量为n个的最小生成树的数量*该最小生成树的树枝的总权值和%mod（题目给出mod值）。对于m条边，连接所有节点的最小生成树只有一个，但是对于m条边，如果不用连接所有节点，那么最小生成树就可能有复数个，并且这些最小生成树的树枝的总权值和是一样的。讲到这应该清楚了，下面看代码吧。（偷偷bb两句：对于这道题的生成器，最小生成树最多只有一个）

#include <stdio.h>
#include<iostream>
#include <algorithm>
#pragma warning(disable:4996)
#define mod 1000000007
#define ll unsigned long long

const int N = 100005;
using namespace std;
int n, m, father[N];
struct node {
	int u, v;//u,v是节点下标
	ll w;//w是连接两个节点的边的权值
}e[N];
int find(int x)
{
	if (x == father[x])
		return father[x];
	return father[x] = find(father[x]);
}
bool cmp(node a, node b)
{
	return a.w < b.w;
}
ll k1, k2;
ll xorShift128Plus()
{
	ll k3 = k1, k4 = k2;
	k1 = k4;
	k3 ^= k3 << 23;
	k2 = k3 ^ k4 ^ (k3 >> 17) ^ (k4 >> 26);
	return k2 + k4;
}
void gen()//generate tree
{
	scanf("%d%d%llu%llu", &n, &m, &k1, &k2);
	for (int i = 1; i <= n; i++)
		father[i] = i;
	for (int i = 1; i <= m; i++)
	{
		e[i].u = xorShift128Plus() % n + 1;
		e[i].v = xorShift128Plus() % n + 1;
		e[i].w = xorShift128Plus();
	}
}
void kruskal()
{
	gen();
	sort(e + 1, e + m + 1, cmp);
	int cnt = 0;//确定操作次数
	ll sum = 0;//求最小生成树的权值和
	for (int i = 1; i <= m; i++)
	{
		int fu = find(e[i].u);
		int fv = find(e[i].v);
		if (fu != fv)//连接树枝
		{
			father[fu] = fv;
			sum += e[i].w;
			cnt++;
			sum %= mod;
		}
		if (cnt == n - 1)//连接n个节点的树已经建好了
			break;
	}
	if (cnt == n - 1)
	{
		printf("%llu\n", sum % mod);
	}
	else
		printf("0\n");
}
int main()
{
	int t;
	scanf("%d", &t);
	while (t--)
	{
		kruskal();
	}
	return 0;
}