python leetcode 474. Ones and Zeroes

最新推荐文章于 2022-12-10 16:10:08 发布

Neekity

最新推荐文章于 2022-12-10 16:10:08 发布

阅读量939

点赞数

分类专栏： python leetcode 文章标签： python leetcode

本文链接：https://blog.csdn.net/Neekity/article/details/84938422

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

leetcode

221 篇文章 2 订阅

订阅专栏

官方提示DP自己用DP做会超时，哈哈哈官方打脸。看了40ms的方法，采用了贪心算法。

按照字符串0或1少的顺序排列
按照取完这个字符串后剩余0或1较多的逆序排列
第2种是对第一种情况的补充例如一个字符串011111111111,0只有一个1很多个
第一种情况对应Array = {“10”, “0001”, “111001”, “1”, “0”}, m = 5, n = 3
第二种情况对应Array = {“01111”, “001”, “100”}, m = 4, n = 4

class Solution:
    def findMaxForm(self, strs, m, n):
        """
        :type strs: List[str]
        :type m: int
        :type n: int
        :rtype: int
        """
        counter = [[i.count('0'), i.count('1')] for i in strs]
        s1 = sorted(counter,key=lambda x:min(x[0],x[1]))
        s2 = sorted(counter,key=lambda x:min(m-x[0],n-x[1]),reverse=True)
        return max(self._findMaxForm(s1, m, n),self._findMaxForm(s2, m, n))

    def _findMaxForm(self, s, m, n):
        count = 0
        for k in s:
            if m >= k[0] and n >= k[1]:
                count += 1
                m -= k[0]
                n -= k[1]
        return count