笔试真题1: n个整数的无序数组，找到每个元素后面比它大的第一个数，要求时间复杂度为O(N)

最新推荐文章于 2021-07-18 10:49:14 发布

黑夜中坚持

最新推荐文章于 2021-07-18 10:49:14 发布

阅读量1.1k

点赞数 1

分类专栏：算法面试文章标签：算法 python leetcode

本文链接：https://blog.csdn.net/qq_26496077/article/details/104353409

版权

算法面试专栏收录该内容

15 篇文章 0 订阅

订阅专栏

本文介绍一种在O(N)时间复杂度内，寻找数组中每个元素后方第一个更大元素的方法。通过使用栈数据结构，算法能高效地处理无序整数数组，返回一个新数组，其中每个元素对应原数组中该位置元素后方的第一个更大的数。

摘要由CSDN通过智能技术生成

n个整数的无序数组，找到每个元素后面比它大的第一个数，要求时间复杂度为O(N)

class Solution:
    def ss(self, l):
        len_l = len(l)
        if len_l == 0:
            return l
        # 栈
        array = []
        # 返回的数组
        res = l
        i = 0
        while i < len_l:
            # 如果当前遍历值比栈顶值小或者栈为空，则将当前遍历值放入栈中
            if len(array) == 0 or l[i] <= l[array[-1]]:
                array.append(i)
                i += 1
            # 如果当前遍历值比栈顶元素大，则递归删除比当前遍历值小的元素，并将遍历值放入栈中及将遍历值放入返回数组中适当位置
            else:
                res[array[-1]] = l[i]
                array.pop()
        # 栈中还不为空，说明没有值比它们大，则将
        while len(array) != 0:
            res[array[-1]] = 2147483647
            array.pop()
        return res


if __name__ == '__main__':
    s = Solution()
    input_values = list(map(int, input().strip().split()))
    result = s.ss(input_values)
    print(result)