349. 兩個數組的交集
給定兩個數組,編寫一個函數來計算它們的交集。
示例 1:
輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2]
示例 2:
輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [9,4]
說明:
輸出結果中的每個元素一定是唯一的。
我們可以不考慮輸出結果的順序。
Code
func intersection(nums1 []int, nums2 []int) []int {
set := make(map[int]bool)
res := make([]int, 0)
for _, v1 := range nums1 {
set[v1] = true
}
// [1,2,2,1]
// [2,2]
// 用hash的value來進行true, false判斷。值得借鑑
for _, v2 := range nums2 {
if true_or_false, ok := set[v2]; ok && true_or_false { //nums2裏面包含nums1裏的元素
res = append(res, v2)
set[v2] = false //防止重複輸出
}
}
return res
}