【GO语言学习】每日一题：根据身高重建队列

题目描述

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对 (h, k) 表示，其中 h 是这个人的身高，k 是应该排在这个人前面且身高大于或等于 h 的人数。 例如：[5,2] 表示前面应该有2 个身高大于等于 5 的人，而 [5,0] 表示前面不应该存在身高大于等于 5 的人。

编写一个算法，根据每个人的身高 h 重建这个队列，使之满足每个整数对 (h, k) 中对人数 k 的要求

示列：

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

解题思路

首先对数组进行排序，第一个元素从高到低，如果第一个元素相同，第二个元素从高到底。即高个子先站前面，低个子再根据切片第二个值来插入到正确的位置。

接着进行如下操作来将低个子插入正确位置

代码展示

type S [][]int

func sortS(s S){
	sort.Sort(s)
}

func (s S)Swap(i,j int) {
	s[i],s[j]=s[j],s[i]
}

func (s S)Len() int{
	return len(s)
}

func (s S)Less(i,j int) bool{
	// 高的人先排，同样高的k值小的在前面
	return s[i][0]>s[j][0] || (s[i][0]==s[j][0] && s[i][1]<s[j][1])
}

func reconstructQueue(people [][]int) [][]int {

	var res [][]int
	sortS(people)//先排序
	for key,value:=range people {
		if value[1]<key {
			//插入index位置
			index:=value[1]
			temp:=append([][]int{},res[index:]...)
			res=append(res[0:index],value)
			res=append(res,temp...)
		}else{
			res=append(res,value)//直接插入res尾部
		}
	}
	return res
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/queue-reconstruction-by-height