算法提高之LeetCode刷题LeetCode Python算法

496. Next Greater Element I

2018-07-14  本文已影响0人  fred_33c7

原题地址:https://leetcode.com/problems/next-greater-element-i/description/

大意:依次找nums1里面的值,如果nums2的面该值后面还有比这个值还大的数,把这个数加到返回的数组中,没有的话,返回-1.

思路:
找下一个,那就next(),比这个数大,比较一下,再用上分片,很简单。

class Solution:
    def nextGreaterElement(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        r_list = []
        for item in nums1:
            r_list.append(next((j for j in nums2[nums2.index(item):] if j > item),-1))
        return r_list

四行代码就能解决问题。

知识点:

上一篇 下一篇

猜你喜欢

热点阅读