小欧皇 | 并查集 联通块 Go

小欧皇

https://www.nowcoder.com/practice/6afe13c512d44b30b03933471d259ba4

题意

小欧正在扮演一个中世纪的皇帝。地图上有n个城市,其中有m条道路,每条道路连接了两个城市。

小欧占领了其中一些城市。如果两个城市可以通过若干条道路互相到达,且这些道路经过的城市都是小欧占领的,那么这两个城市之间就可以通过经商获得收益1。请注意,每两个城市之间的收益只会被计算一次。

现在,小欧准备占领一个未被占领的城市,使得总收益最大化。你能帮帮她吗?

思路

如果一个联通块的大小为x,那么可以获得的收益是C[x][2] ,也就是从这个联通块里选2个点的方案数,x*(x-1)/2

具体思路:

  1. 用并查集找出所有的联通块和联通块大小
  2. 维护当前能够得到的收益
  3. 枚举所有未占领的城市,看把这个城市占领后能得到的收益。具体的计算过程是维护新联通块的大小,枚举所有和这个点相邻的点,减去这个点所在的联通块的贡献,再将这个联通块的大小加入到新联通块的大小里。这里要注意去重,每个联通块只需计算一次,可以用哈希表标识下每个root节点

Go代码

package main

import (
	"fmt"
)

func main() {
	var n, m int
	fmt.Scan(&n, &m)
	var s string
	fmt.Scan(&s)
	s = " " + s //下标从1开始
	grid := make([][]int, n+1)
	root := make([]int, n+1) //祖先节点
	siz := make([]int, n+1)  //联通块大小
	vis := make([]bool, n+1)
	for i := 1; i <= n; i++ {
		grid[i] = make([]int, 0)
		root[i] = i
		siz[i] = 1
		vis[i] = false
	}
	var find func(int) int
	find = func(x int) int {
		if x != root[x] {
			root[x] = find(root[x])
		}
		return root[x]
	}
	merge := func(x, y int) {
		fx, fy := find(x), find(y)
		if fx == fy {
			return
		}
		root[fx] = fy
		siz[fy] += siz[fx]
	}
	for i := 1; i <= m; i++ {
		var u, v int
		fmt.Scan(&u, &v)
		if s[u] == '1' && s[v] == '1' {
			merge(u, v)
		}
		grid[u] = append(grid[u], v)
		grid[v] = append(grid[v], u)
	}
	//计算初始总贡献
	sum := 0
	for i := 1; i <= n; i++ {
		if s[i] == '0' {
			continue
		}
		fa := find(i)
		if vis[fa] {
			continue
		}
		vis[fa] = true
		sum += siz[fa] * (siz[fa] - 1) / 2
	}
	ansVal, ansIdx := 0, 0
	for i := 1; i <= n; i++ {
		if s[i] == '1' {
			continue
		}
		tmpSum, nowCnt := sum, 1
		mp := make(map[int]int)
		for _, ne := range grid[i] {
			if s[ne] == '0' {
				continue
			}
			fne := find(ne)
			if _, ok := mp[fne]; ok {
				continue
			}
			mp[fne] = 1
			tmpSum -= siz[fne] * (siz[fne] - 1) / 2 //减去原先联通块的贡献
			nowCnt += siz[fne]
		}
		tmpSum = tmpSum + nowCnt*(nowCnt-1)/2 //加上新联通块数量
		if ansVal < tmpSum {
			ansVal = tmpSum
			ansIdx = i
		}
	}
	fmt.Printf("%d %d",  ansIdx,ansVal)
}

#牛客创作赏金赛#
15天大厂真题带刷Go题解 文章被收录于专栏

15天大厂真题带刷Golang题解

全部评论

相关推荐

04-12 21:52
南开大学 Java
鼠鼠有点摆,去年边学着没敢投简历,没实习。从1月到现在总共面了五次,四次字节的日常(HR打电话约面试才敢去的),然后一次腾讯的暑期,都是一面挂,其他则是没给面。暑期的岗,4.2才开始海投,前面想着等字节第四次一面后再投,结果挂,而且感觉投晚了。字节投了11个,9个简历挂,剩下2个没动静。阿里全都简历挂,剩下的在&quot;投递简历&quot;。腾讯给了一次面。然后其他大中厂、手机厂什么的都是做完测评or笔试就没下文,打开几个看也是终止流程,感觉剩下的也应该是简历挂了。感觉是简历的原因?项目部分,几次面试,感觉面试官主要就拷问过秒杀这一个点。自己说的时候会尝试把sse那条说成亮点,但除了腾讯面试官问过一下这整个点在业务方面对用户有什么用之类的问题外,其他最多只是问一下sse八股...感觉也许不是很让面试官感兴趣。这个短链接也是无人问津,就被问过一回雪花算法的设计。也许我该拿点评改改,然后再在网上找一个什么项目,凑两个,而不是用自己现在这两个项目?或者是点评改改放前面,然后原本第一个项目,把秒杀抽掉,剩下的想办法从网上火的RAG项目里移植点亮点,或者直接就用网上的RAG项目?感觉我主要还是偏向后端开发,但是感觉如果除开点评,再拿一个项目,想不到有什么自己能掌控且跟点评不重的。然后鼠鼠之前主要的问题是担心面试让打开项目演示,然后就一直花时间在用AI整第一个项目,第二个项目都没时间整,第四次面试之前还因为太害怕被认为不熟悉项目,跟AI一起把简历的说辞做了大幅度弱化,然后暑期都是拿弱化后的简历投的,感觉是不是看上去太没有吸引力就直接给简历挂了。(图1是弱化后的,图2是弱化前的,但之前3月初投了几家好像也是简历挂。)而且因为3月花了很多时间整在跟AI整代码,导致八股和算法都没怎么看,算法之前有跟灵神题单刷一些,还算入门,但是八股只看了一些基本的,可能面试的时候只答得上来60-70%,而且表述有些混乱,都是想到哪说到哪;前面几回面试基本上都有大板块的基础八股没答出来,比如RedisZ&nbsp;Set数据结构,MQ延时消息、可靠性保证,JVM内存分配的过程、GC&nbsp;roots,JUC锁,设计模式。现在有点不知道该怎么办。求大佬们给点简历修改建议或者面试准备建议,不胜感激!
何时能不做牛马:简历每个点之间的间距可以缩一下。几乎没遇到过要演示项目的情况,即使万一遇上了你也可以说部署在其他电脑上本地没代码。nku不应该简历挂吧?抓紧背背八股练练表达,不要放弃,五六月份找到也不晚(不然还得提前入职
应届生简历当中,HR最关...
点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

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