Golang Leetcode 350. Intersection of Two Arrays II.go

思路

和上一个求交集不同的是，有重复元素，所以在map保存的时候，需要保存一个出现的次数

code

func intersect(nums1 []int, nums2 []int) []int {
	m := make(map[int]int)

	for _, v := range nums1 {
		if _, ok := m[v]; ok {
			m[v]++
		} else {
			m[v] = 1
		}
	}

	ret := []int{}

	for _, v := range nums2 {
		if val, ok := m[v]; val > 0 && ok {
			ret = append(ret, v)
			m[v]--
		}
	}
	return ret
}

更多内容请移步我的repo：https://github.com/anakin/golang-leetcode