leetcode算法题--元素值大于变化阈值的子数组

最新推荐文章于 2024-08-22 17:27:46 发布

bob62856

最新推荐文章于 2024-08-22 17:27:46 发布

阅读量143

点赞数

分类专栏： Algorithm 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/qq_20817327/article/details/132220305

版权

Algorithm 专栏收录该内容

374 篇文章 3 订阅

订阅专栏

本文介绍了一种使用并查集数据结构解决LeetCode问题的方法，目标是找到满足特定条件下元素大于阈值的子数组的最大可能大小。算法首先对数组进行排序，然后利用并查集合并相同元素，统计符合条件的子数组长度。

摘要由CSDN通过智能技术生成

原题链接：https://leetcode.cn/problems/subarray-with-elements-greater-than-varying-threshold/description/

func validSubarraySize(nums []int, threshold int) int {
	// 并查集模版 
	n := len(nums)
	fa := make([]int, n+1)
	for i := 0; i <= n; i ++ {
		fa[i] = i
	}
	var find func(x int) int
	find = func(x int) int {
		if fa[x] != x {
			fa[x] = find(fa[x])	
		}
		return fa[x]
	}	
	type pair struct {
		i int
		v int
	}
	pairs := make([]pair, 0)
	for i, num := range nums {
		pairs = append(pairs, pair{i, num})
	}
	sort.Slice(pairs, func(i, j int) bool {
		if pairs[i].v > pairs[j].v {
			return true
		}
		return false
	})
	sz := make([]int, n+1)
	for _, pair := range pairs {
		i := pair.i			
		j := find(i+1)
		fa[i] = j // 合并i和i+1 
		sz[j] += sz[i] + 1
		// fmt.Println(fa, sz)
		if pair.v > threshold/sz[j] {
			return sz[j]
		} 
	}
	return -1
}