标签：node hdu int Tree vex fib 4786 n2 mins

题目链接：hdu4786

题意：给你一个图，判断是否存在一个生成树，使得权值之和为一个斐波那契数。

思路：分别跑一次最大生成树和最小生成树。然后判断两个生成树的权值中间是否存在一个斐波那契数。如果存在则输出yes，否则输出no

代码：

#include<bits/stdc++.h>
using namespace std;

const int maxn = 1e5 + 10;

struct node {
	int from, to, w;
	node(int a, int b, int c) :from(a), to(b), w(c) {}
};

int vex[maxn];
int n, m;

int fib[maxn];
vector<node> v;

void fibs() {
	int i;
	fib[0] = 1;
	fib[1] = 1;
	for (i = 2; i < 100000; i++)
		fib[i] = fib[i - 1] + fib[i - 2];
}


void init() {
	int i;
	for (i = 0; i <= n; i++) {
		vex[i] = i;
	}
}

int find(int x)
{
	if (vex[x] == x)
		return x;
	return vex[x] = find(vex[x]);
}


void kruskal(int &cnt) {
	int i;
	int a, b;
	for (i = 0; i < m; i++) {
		auto e = v[i];
		a = e.from;
		b = e.to;
		int r1 = find(a);
		int r2 = find(b);
		if (r1 != r2) {
			vex[r1] = r2;
			cnt += e.w;
		}
	}
}

int main() {
	ios::sync_with_stdio(false);
	fibs();
	int t;
	cin >> t;
	int cases = 0;
	while (t--) {
		v.clear();
		cases++;
		cin >> n >> m;
		int a, b, w, i;
		for (i = 0; i < m; i++) {
			cin >> a >> b >> w;
			v.push_back(node(a, b, w));
		}
		sort(v.begin(), v.end(), [=](node n1, node n2)
			{
				return n1.w > n2.w;
			});
		memset(vex, 0, sizeof(vex));
		int flag = 0;
		int maxs, mins;
		maxs = mins = 0;
		init();
		kruskal(maxs);
		memset(vex, 0, sizeof(vex));
		sort(v.begin(), v.end(), [=](node n1, node n2)
			{
				return n1.w < n2.w;
			});
		init();
		kruskal(mins);
		int cnts = 0;

		for (i = 0; maxs >= fib[i]; i++) {
			if (mins <= fib[i] && maxs >= fib[i]) {
				flag = 1;
				break;
			}
		}
		for (i = 1; i <= n; i++) {
			if (vex[i] == i)
				cnts++;
			if (cnts > 1)
				break;
		}

		if (cnts > 1)
			flag = 0;
		cout << "Case #" << cases << ": ";
		if (flag)
			cout << "Yes" << endl;
		else
			cout << "No" << endl;
	}
	return 0;
}

• 点赞
• 收藏
• 分享
• • 文章举报

jiangxiaoju 发布了34 篇原创文章 · 获赞 44 · 访问量 3万+ 私信 关注

标签：node,hdu,int,Tree,vex,fib,4786,n2,mins
来源： https://blog.csdn.net/qq_43058685/article/details/104126473

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。