LeetCode 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 处的学生高度不匹配。

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

对数组排序,然后对比排序前后数组有几个对应位置的元素不同:

class Solution {
public:
    int heightChecker(vector<int>& heights) {
        vector<int> heightsBak = heights;
        
        sort(heights.begin(), heights.end());

        int ans = 0;
        for (int i = 0; i < heights.size(); ++i) {
            if (heightsBak[i] != heights[i]) {
                ++ans;
            }
        }

        return ans;
    }
};

如果输入数组长度为n,此算法时间复杂度为O(nlgn),空间复杂度为O(1)。

题目中规定了1 <= heights[i] <= 100,因此可用计数排序将时间复杂度降为O(n):

class Solution {
public:
    int heightChecker(vector<int>& heights) {
        vector<int> countArr(100);
        for (int i : heights) {
            ++countArr[i - 1];
        }

        int index = 0;
        int ans = 0;
        for (int i = 0; i < 100; ++i) {
            if (!countArr[i]) {
                continue;
            }

            while (countArr[i]) {
                if (heights[index] != i + 1) {
                    ++ans;
                }

                ++index;
                --countArr[i];
            }
        }
        
        return ans;
    }
};

如果高度范围为m,输入数组长度为n,此算法时间复杂度为O(n),空间复杂度为O(m)。

posted @ 2023-02-01 21:47  epiphanyy  阅读(21)  评论(0)    收藏  举报  来源