两个数组交集与合并两个有序数组

leetcode 350. 两个数组的交集 II

• 排序

func intersect(nums1 []int, nums2 []int) []int {
	// 排序
	sort.Ints(nums1)
	sort.Ints(nums2)
	var i, j int
	var ret []int
	for i < len(nums1) && j < len(nums2) {
		// 一个比另一个大，就把另一个指针往后移
		if nums1[i] > nums2[j] {
			j++
		} else if nums1[i] < nums2[j] {
			i++
		} else {
			ret = append(ret, nums1[i])
			i++
			j++
		}
	}
	return ret
}

• 哈希表

func intersect(nums1 []int, nums2 []int) []int {
	var ret []int
	m := map[int]int{}
	for _, i := range nums1 {
		m[i]++
	}
	for _, i := range nums2 {
		// 该数在A数组中数量大于0
		if m[i] > 0 {
			// 将该数数量减小1
			m[i]--
			// 把该数放到结果集
			ret = append(ret, i)
		}
	}
	return ret
}

leetcode 88. 合并两个有序数组

func merge(nums1 []int, m int, nums2 []int, n int) {
	var i, j int
	for i < len(nums1) && j < len(nums2) {
		if nums1[i] >= nums2[j] {
			// 挪出一个空位
			for a := m + j; a > i; a-- {
				nums1[a], nums1[a-1] = nums1[a-1], nums1[a]
			}
			// 放j进去
			nums1[i] = nums2[j]
			// 去判断下一个j
			j ++
		}
		i ++
	}
	// j剩下的都大，直接放
	if j < len(nums2) {
		for k := j; k < len(nums2); k++ {
			nums1[m+k] = nums2[k]
		}
	}
	return
}

• 先放再排序

func merge(nums1 []int, m int, nums2 []int, n int) {
	for i := 0; i < len(nums2); i++ {
		nums1[m+i] = nums2[i]
	}
	sort.Ints(nums1)
	return
}

• 最优解

func merge(nums1 []int, m int, nums2 []int, n int) {
	// p1 是nums1数据的末尾，p2是nums2末尾，p是nums1数组的末尾
	p1, p2, p := m-1, n-1, m+n-1
	for p1 >= 0 && p2 >= 0 {
		// 哪个大，就把他放到p位置，然后往前推
		if nums1[p1] >= nums2[p2] {
			nums1[p] = nums1[p1]
			p1--
		} else {
			nums1[p] = nums2[p2]
			p2--
		}
		p--
	}
	// 若p2 == -1, 此句无效 
    // 若p1 == -1, nums2前p2个数是最小的，复制到nums1中
    copy(nums1[:p2+1],  nums2)
}