【Golang語言】LeetCode1122. Relative Sort Array

給你兩個數組，arr1 和 arr2， arr2 中的元素各不相同 arr2 中的每一個元素都出如今 arr1 中 對 arr1 中的元素進行排序，使 arr1 中項的相對順序和 arr2 中的相對順序相同。未在 arr2 中出現過的元素須要按照升序放在 arr1 的末尾。 leetcode-cn.com/problems/re…

輸入：arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
輸出：[2,2,2,1,4,3,3,9,6,7,19]
複製代碼

題意：arr2給你自定義了一個數字大小順序，將arr1裏的數按照2中定義的規則拍一拍。

思路：統計1中的數字出現的次數，遍歷2，依次取出1中的數，按照統計的次數追加，最後把沒出如今2中的加到最後。

func relativeSortArray(arr1 []int, arr2 []int) []int {
	maps := make(map[int]int, 0)
	result := make([]int, 0)
	for _, v := range arr1 {
		maps[v]++
	}
	for _, v := range arr2 {
		if times, ok := maps[v]; ok {
			for i := 0; i < times; i++ {
				result = append(result, v)
			}
			delete(maps, v)
		}
	}
	keys := make([]int, 0)
	for v := range maps {
		keys = append(keys, v)
	}
	sort.Ints(keys)
	for _, v := range keys {
		times := maps[v]
		for i := 0; i < times; i++ {
			result = append(result, v)
		}
	}

	return result
}

複製代碼