小欧皇 | 并查集 联通块 Go
小欧皇
https://www.nowcoder.com/practice/6afe13c512d44b30b03933471d259ba4
题意
小欧正在扮演一个中世纪的皇帝。地图上有n个城市,其中有m条道路,每条道路连接了两个城市。
小欧占领了其中一些城市。如果两个城市可以通过若干条道路互相到达,且这些道路经过的城市都是小欧占领的,那么这两个城市之间就可以通过经商获得收益1。请注意,每两个城市之间的收益只会被计算一次。
现在,小欧准备占领一个未被占领的城市,使得总收益最大化。你能帮帮她吗?
思路
如果一个联通块的大小为x,那么可以获得的收益是C[x][2] ,也就是从这个联通块里选2个点的方案数,x*(x-1)/2
具体思路:
- 用并查集找出所有的联通块和联通块大小
- 维护当前能够得到的收益
- 枚举所有未占领的城市,看把这个城市占领后能得到的收益。具体的计算过程是维护新联通块的大小,枚举所有和这个点相邻的点,减去这个点所在的联通块的贡献,再将这个联通块的大小加入到新联通块的大小里。这里要注意去重,每个联通块只需计算一次,可以用哈希表标识下每个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题解