leetcode 406. 根据升高重建队列 golang实现

描述

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。

注意：
总人数少于1100人。

示例

输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

思路

1. 按照身高从大到小排序， 相同身高按k从小到大排序
2. 遍历数组 由于每次遍历的人的身高都<=前面的人的身高，所以每次只需要看这个前面有几个人 即把这个人插入到新数组的第几个位置即可

如:
排序后: [[7,0],[7,1],[6,1],[5,0],[5,2],[4,4]]
7,0 插入0位置 [[7,0]]
7,1 插入1位置 [[7,0],[7,1]]
6,1 插入到1 位置 [[7,0],[6,1],[7,1]]
5,0 插入到0位置 [[5,0],[7,0],[6,1],[7,1]]
5,2 插入到2 位置 [[5,0],[7,0],[5,2],[6,1],[7,1]]
4,4 插入到4位置 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]

实现

type PeopleSlice [][]int

func (s PeopleSlice) Len() int      { return len(s) }
func (s PeopleSlice) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
func (s PeopleSlice) Less(i, j int) bool {
	if s[i][0] > s[j][0] {
		return true
	} else if s[i][0] == s[j][0] && s[i][1] < s[j][1] {
		return true
	}else{
		return false
	}
}
func reconstructQueue(people [][]int) [][]int {
	var ret = make([][]int, 0)

	sort.Sort(PeopleSlice(people))
	for _, p := range people{
		tail := append([][]int{}, ret[p[1]:]...)
		ret = append(ret[0:p[1]], p)
		ret = append(ret, tail...)
	}

	return ret
}