【计数排序】高度检查器

题目

1051. 高度检查器

---

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

排序后的高度情况用整数数组 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

解题

方法一：排序

思路

题目翻译成人话就是求排序前和排序后位置不一样的个数。

直接模拟。

代码

class Solution {
    public int heightChecker(int[] heights) {
        int[] expected = heights.clone();
        Arrays.sort(expected);
        int count = 0;
        for (int i = 0; i < heights.length; i++) {
            if (heights[i] != expected[i]) count++;
        }
        return count;
    }
}

优化：计数排序

由于数据范围只有 $10^2$ ，我们可以使用时间复杂度最优秀仅有 $O(N)$ 的计数排序来解决。

class Solution {
    public int heightChecker(int[] heights) {
        int[] helper = new int[101];
        for (int height : heights) helper[height]++;
        int curr = 0, count = 0;
        for (int i = 1; i < 101; i++) {
            while (helper[i]-- > 0) {
                if (heights[curr++] != i) count++;
            }
        }
        return count;
    }
}