题解 | #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 ; 
}
全部评论

相关推荐

05-12 10:10
已编辑
门头沟学院 人工智能
写这篇之前我犹豫了挺久。一方面是怕被人骂,&quot;又一个收割焦虑的转行帖&quot;;另一方面是看了太多用&nbsp;GPT&nbsp;套娃出来的「学习路线」文章,AI&nbsp;味重得让人没法读完。所以这篇全是亲身踩过的坑,时间线、用过的项目、当时的心路全都尽量原样写出来。如果你是大学生在迷茫要不要转&nbsp;AI,或者已经在转的路上,希望能给点参考。&nbsp;一个反共识的开场:你以为进&nbsp;OpenAI&nbsp;的人都是博士?&nbsp;先讲个故事,跟我没关系,但跟所有想转&nbsp;AI&nbsp;的人都有关系。&nbsp;OpenAI&nbsp;的&nbsp;Sora&nbsp;团队(就是搞文生视频那个)一共&nbsp;13&nbsp;个人。这里面有两个人特别有意思:&nbsp;Will&nbsp;DePue,密歇根大学计算机系,直接辍学了。17...
_hengheng:我也本,也算是做ai相关,我最开始感觉做ai工程师有多么多么困难,后来发现懂了原理后整体训练完全可以看成一个流程化的内容,开源方案太多了,大多基本都是按着模子在自家业务上做各种操作,就算是大厂的小部门也没那么多资源去训基模,反而更多的是像怎么把技术往业务方向靠近了,不过当前时代如果本科学历没那么好加上自己执行力不是特别强还真不建议走ai工程师这条路,可以试试其他ai的偏业务方向,不然校招不太好杀出来
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

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