【题目描述】
学校在拍年度纪念照时,一般要求学生按照 非递减 的高度顺序排列。
请你返回能让所有学生以 非递减 高度排列的最小必要移动人数。
注意,当一组学生被选中时,他们之间可以以任何可能的方式重新排序,而未被选中的学生应该保持不动。
示例:
输入:heights = [1,1,4,2,1,3]
输出:3
解释:
当前数组:[1,1,4,2,1,3]
目标数组:[1,1,1,2,3,4]
在下标 2 处(从 0 开始计数)出现 4 vs 1 ,所以我们必须移动这名学生。
在下标 4 处(从 0 开始计数)出现 1 vs 3 ,所以我们必须移动这名学生。
在下标 5 处(从 0 开始计数)出现 3 vs 4 ,所以我们必须移动这名学生。
示例 2:
输入:heights = [5,1,2,3,4]
输出:5
示例 3:
输入:heights = [1,2,3,4,5]
输出:0
提示:
1 <= heights.length <= 100
1 <= heights[i] <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/height-checker
【解题思路】
首先要注意的是这道题说:当一组学生被选中时,他们之间可以以任何可能的方式重新排序,而未被选中的学生应该保持不动。所以题目的意思是找出来有哪几个人需要换位置即可。那么首先想到的就是排序,排序后的数组和当前的数组之间不同的数字的个数就是需要换位置的人,这样就需要O(nlogn)的时间复杂度。
但是这道题有一种时间复杂度为O(n)的解法,用一个count数组记录每个数字出现的次数。然后遍历count数组,当count[i]==当前数组的第一个元素,就表示这个人是不需要换位置的,当count[i]!=当前数组的第一个元素,就表示这个人需要换位置,res就需要加1,不管用不用换位置count[i]都应该减1,直到count[i]=0的时候换下一个count[i+1]继续重复上述步骤。用Python实现的代码如下:
class Solution(object):
def heightChecker(self, heights):
"""
:type heights: List[int]
:rtype: int
"""
count = [0]*101
for item in heights:
count[item] += 1
res = 0
for i in range(1, 101):
while count[i]>0:
item = heights.pop(0)
if item != i:
res += 1
count[i] -= 1
return res