2024-05-01:用go语言,给定两个长度为偶数n的整数数组nums1和nums2, 分别移除它们各自的一半元素, 将剩下的元素合并成集合s。 找出集合s中可能包含的最多元素数量。 输入:nums
2024-05-01:用go语言,给定两个长度为偶数n的整数数组nums1和nums2,
分别移除它们各自的一半元素,
将剩下的元素合并成集合s。
找出集合s中可能包含的最多元素数量。
输入:nums1 = [1,2,3,4,5,6], nums2 = [2,3,2,3,2,3]。
输出:5。
答案2024-05-01:
题目来自leetcode3002。
大体步骤如下:
1.创建两个空的布尔型map,分别为set1和set2,用于存储nums1和nums2中的元素。
2.遍历nums1,将元素添加到set1中,以便记录每个元素的出现情况。
3.遍历nums2,将元素添加到set2中,同样记录每个元素的出现情况。
4.记录两个数组的交集元素数量,这里用common表示。
5.获取set1和set2中各自不同元素的数量,分别为n1和n2。
6.初始化答案ans为n1 + n2 - common,即为合并后的集合s中可能包含的最多元素数量。
7.计算移除元素的数量m(即数组长度的一半)。
8.如果set1中的元素数量大于m,则进入条件判断:
-
找出需要移除的元素数量(mn)为n1 - m和common中较小的值。
-
更新答案ans,减去需要移除的元素数量。
-
更新common,减去移除的数量mn。
9.同样处理set2中的元素:
-
如果set2中的元素数量大于m,则继续进行下一步操作。
-
更新n2,减去需要移除的元素数量,确保集合s的大小不超过m。
-
更新答案ans,相应地减去多余的元素数量。
10.返回最终的答案ans。
总的时间复杂度为O(n),其中n表示nums1和nums2的总长度。
总的额外空间复杂度是O(n),主要用于存储set1和set2的元素。
Go完整代码如下:
package main
import (
"fmt"
)
func maximumSetSize(nums1, nums2 []int) int {
set1 := map[int]bool{}
for _, x := range nums1 {
set1[x] = true
}
set2 := map[int]bool{}
for _, x := range nums2 {
set2[x] = true
}
common := 0
for x := range set1 {
if set2[x] {
common++
}
}
n1 := len(set1)
n2 := len(set2)
ans := n1 + n2 - common
m := len(nums1) / 2
if n1 > m {
mn := min(n1-m, common)
ans -= n1 - mn - m
common -= mn
}
if n2 > m {
n2 -= min(n2-m, common)
ans -= n2 - m
}
return ans
}
func min(a, b int) int {
if a < b {
return a
}
return b
}
func main() {
nums1 := []int{1, 2, 3, 4, 5, 6}
nums2 := []int{2, 3, 2, 3, 2, 3}
result := maximumSetSize(nums1, nums2)
fmt.Println(result)
}
Python完整代码如下:
# -*-coding:utf-8-*-
def maximumSetSize(nums1, nums2):
set1 = set(nums1)
set2 = set(nums2)
common = len(set1 & set2)
n1 = len(set1)
n2 = len(set2)
ans = n1 + n2 - common
m = len(nums1) // 2
if n1 > m:
mn = min(n1 - m, common)
ans -= n1 - mn - m
common -= mn
if n2 > m:
n2 -= min(n2 - m, common)
ans -= n2 - m
return ans
def min(a, b):
return a if a < b else b
if __name__ == "__main__":
nums1 = [1, 2, 3, 4, 5, 6]
nums2 = [2, 3, 2, 3, 2, 3]
result = maximumSetSize(nums1, nums2)
print(result)
公众号:福大大架构师每日一题