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 @   epiphanyy  阅读(14)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
历史上的今天:
2022-02-01 MySQL必知必会 学习笔记 第二十章 更新和删除数据
点击右上角即可分享
微信分享提示