题目描述
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对 (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