646. Maximum Length of Pair Chain

问题

给定n对数,每对数里面第一个数比第二个数小。现在定义(c,d)可以跟在(a,b)后面组串,如果b<c。问最多能组多长的串。

Input: [[1,2], [2,3], [3,4]]
Output: 2
Explanation: [1,2] -> [3,4]

思路

当(a,b)和(c,d)组串时,要满足b<c,此时a<b,c<d所以有a<b<c<d。
继续组串的话,要用d去找,找到一个(e,f),满足d<e,而e又小于f,接着根据f去找。
可以看到每次要组新串时候需要根据第二个数去找,为了能让串尽量长,我们要让第二个数尽量小(这样就可以拉很多个)。
所以直觉的想法就是把这些数据对按第二个数从小到大排序,然后从前往后串,能组就组,不能组就跳过,直到遍历完全部。

时间复杂度O(n),空间复杂度O(1)

代码

class Solution(object):
    def findLongestChain(self, pairs):
        """
        :type pairs: List[List[int]]
        :rtype: int
        """
        cur = float('-inf')
        res = 0
        for x, y in sorted(pairs, key= lambda a : a[1]):
            if x > cur:
                res += 1
                cur = y
        return res  

相关知识

float('inf')    # python正无穷,所有数都比它小
float('-inf')   # python负无穷,所有数都比它大

l.sort()        # python内置的排序函数,在本地排序,不返回副本
l2 = sorted(l)  # 返回一个副本,原始输入不变。

l = [ [2,4], [8,16,32,64], [128,256,512]]
sorted(l, key= lambda a : a[1])                    # 按第二维的值进行升序排序
sorted(l, key= lambda a : len(a), reverse=True)    # 按长度进行降序排序
posted @ 2018-10-08 22:29  PilgrimHui  阅读(197)  评论(0编辑  收藏  举报