leetcode-高度检查器

99 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第19天,点击查看活动详情

题目描述

学校打算为全体学生拍一张年度纪念照。根据要求,学生需要按照 非递减 的高度顺序排成一行。

排序后的高度情况用整数数组 expected 表示,其中 expected[i] 是预计排在这一行中第 i 位的学生的高度(下标从 0 开始)。

给你一个整数数组 heights ,表示 当前学生站位 的高度情况。heights[i] 是这一行中第 i 位学生的高度(下标从 0 开始)。

返回满足 heights[i] != expected[i] 的 下标数量 。

示例:
输入:heights = [1,1,4,2,1,3]
输出:3
解释:
高度:[1,1,4,2,1,3]
预期:[1,1,1,2,3,4]
下标 2 、4 、5 处的学生高度不匹配。

示例 2:
输入:heights = [5,1,2,3,4]
输出:5
解释:
高度:[5,1,2,3,4]
预期:[1,2,3,4,5]
所有下标的对应学生高度都不匹配。

示例 3:
输入:heights = [1,2,3,4,5]
输出:0
解释:
高度:[1,2,3,4,5]
预期:[1,2,3,4,5]
所有下标的对应学生高度都匹配。
 

提示:

  • 1 <= heights.length <= 100
  • 1 <= heights[i] <= 100

思路

题意比较容易理解,就是要把学生身高按照从小到大排列,因为可能存在相同的情况,所以表述为非递减。自然而然的想法是,先对数组进行排序,然后再逐个比较找出不一样的数量。这里有个范围限制- 1 <= heights[i] <= 100,范围非常小,我们可以考虑使用桶排序,这样可以把排序的时间复杂度从O(nlogn)降低到O(n)。
另外,这里在遍历桶的时候有个小技巧,可以在for循环的退出条件上加上index < heights.length,避免遍历最后的一些空桶。

Java版本代码

class Solution {
    public int heightChecker(int[] heights) {
        int[] order = new int[101];
        for (int height : heights) {
            order[height]++;
        }
        int ans = 0;
        int index = 0;
        for (int i = 1; i <= 100 && index < heights.length; i++) {
            for (int j = 1; j <= order[i]; j++) {
                if (heights[index++] != i) {
                    ans++;
                }
            }
        }
        return ans;
    }
}