题解 | #Tree#

黑暗城堡

https://ac.nowcoder.com/acm/contest/958/A

Tree

solution

求一棵恰好有 needneed 条白边的最小生成树,根据 MSTMST 的求解方法,一条白边是否在 MSTMST 中只与白边的权值有关。所以我们只需对白边进行增量操作即可。假设我们对图中的每条白边的增量为 xxMSTMST 中白边的数量记为 f(x)f(x) ,可以易看出 f(x)f(x) 的大小是随 xx 的增大而增大的,满足单调性,考虑二分增量。 那么这道题就可以切了。。。

code

using Simon O_O ;
int n , m , need , ans ;
struct node 
{
	int frm , to , dis ;
	bool col ; 
} ed[maxn << 1] ;
bool operator <(node x , node y)
{
	if(x.dis == y.dis)	return x.col <  y.col ;//权值相同优先选白边
	return x.dis < y.dis ;
} 
int dis[maxn] , Cnt[maxn] , fa[maxn] ;
int find(int x)
{
	if(x == fa[x])	return x ; 
	return fa[x] = find(fa[x]) ; 
}
int kru(int x)
{
	int tot = 0 ;	ans = 0 ; 
	for(int i = 1 ; i <= m ; i++ )
		if(!ed[i].col)	ed[i].dis += x ;
	fa[0] = 0 ; 
	for(int i = 1 ; i <= n ; i++ )	fa[i] = i ;                                         
	sort(ed + 1 , ed + m + 1) ;
	for(int i = 1 ; i <= m ; i++ )
	{
		int u = find(ed[i].frm) , v = find(ed[i].to) ; 
		if(u == v)	continue ; 
		fa[v] = u ; 
		if(!ed[i].col)	tot++ ; 	ans += ed[i].dis ; 		
	}
	for(int i = 1 ; i <= m ; i++ )
		if(!ed[i].col)	ed[i].dis -= x ;
	return tot ;//统计MST中白边个数
}
signed main()
{
	n = read() , m = read() , need = read() ;
	int mid = 0 ;
	for(int i = 1 ; i <= m ; i++ )
	{
		int u , v , w , col ; 
		u = read() , v = read() , w = read() , col = read() ;
		ed[i] = {u , v , w , col}  ; 
	}
	int l = -105 , r = 105 ; 
	int ans2 = 0 ; 
	while(l < r)
	{
		mid = l + r >> 1 ;
		if(kru( mid ) >= need )		l = mid + 1 , ans2 = ans - need * mid ; 
		else r = mid ; 
	}
	cout << ans - need * mid ; 
	return 0 ; 
}
全部评论

相关推荐

01-28 16:12
中南大学 Java
几年前还没有chatgpt的时候,刷题真的是很痛苦。刷不出来只能看题解,题解有几个问题:第一个是每次看的写题解的人都不一样,很难有一个统一的思路;第二个也是最重要的是,题解只提供了作者自己的思路,但是没有办法告诉你你的思路哪里错了。其实很少有错误的思路,我只是需要被引导到正确的思路上面去。所以传统题解学习起来非常困难,每次做不出来难受,找题解更难受。但是现在chatgpt能做很多!它可以这样帮助你&nbsp;-1.&nbsp;可以直接按照你喜欢的语言生成各种解法的题解和分析复杂度。2.&nbsp;把题和你写的代码都发给它,它可以告诉你&nbsp;你的思路到底哪里有问题。有时候我发现我和题解非常接近,只是有一点点🤏想错了。只要改这一点点就是最优解。信心倍增。3.&nbsp;如果遇到不懂的题解可以一行一行询问为什么要这样写,chatgpt不会嫌你烦。有时候我觉得自己的range写错了,其实那样写也没错,只是chat老师的题解有一点优化,这个它都会讲清楚。4.&nbsp;它可以帮你找可以用同类型解法来做的题。然后它可以保持解法思路不变,用一个思路爽刷一个类型的题。如果题目之间思路又有变化,它会告诉你只有哪里变了,其他的地方还是老思路。5.&nbsp;它也可以直接帮你总结模板,易错点。经过chat老师的指导,我最大的改变是敢刷题了。之前刷题需要先找某一个人写的算法题repo,然后跟着某一个人他的思路刷他给的几个题。如果想写别的题,套用思路失败了,没有他的题解,也不知道到底哪里错了;看别人的题解,思路又乱了。这个问题在二分查找和dp类型的题里面特别常见。但是现在有chat老师,他会针对我的代码告诉我我哪里想错了,应该怎么做;还按照我写代码的习惯帮我总结了一套属于我的刷题模板。每天写题全是正反馈!
明天不下雨了:那我建议可以用 chatgpt atlas 或者 dia 去刷,也可以用 chrome 加个 ai 插件去刷 左边刷题右边 chat 效果很好
AI时代的工作 VS 传...
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务