Leetcode Weekly Contest 174

5328. The K Weakest Rows in a Matrix

1. 题意

给定一个二维矩阵mat，每一行由0,1组成， 1代表士兵，0代表平民，如果一行士兵更多，则称这一行的战斗力更强，求从弱到强的k行，返回下标。

2. 思路

简单的统计每一行中1的个数，然后进行排序即可
这里不仅要记录每一行中1的个数，还得记录每一行的下标！！！
这里不能使用元组来做，因为元组不支持赋值操作，也就是不能进行修改！

第一种方法：通过dict来存储， 浪费存储空间了！
这里得学会对dict进行排序，排序后返回了一个元组的数组
c = sorted(cc.items(), key = lambda d:d[1]) # 变成元组了啊！

第二种方法：通过二维数组来存储！
cc.sort(key = lambda c:c[1]) # 数组按照第二项进行从小往大排序

3. python实现

class Solution(object):
    def kWeakestRows(self, mat, k):
        """
        :type mat: List[List[int]]
        :type k: int
        :rtype: List[int]
        """
# ----------------刚刚比赛时写的并不好啊1------------------------------------
#         cc = collections.defaultdict(int)
#         for i in range(len(mat)):
#             for j in range(len(mat[0])):
#                 if mat[i][j] == 1:
#                     cc[i] += 1
#                 else:
#                     if j==0:
#                         cc[i] = 0
#                     break
        
#         c = sorted(cc.items(), key = lambda d:d[1]) # 变成元组了啊！
#         # print(c)
#         res = []
#         for nums in c:
#             if k > 0:
#                 res.append(nums[0])
#             k -= 1
#         return res
        
        m, n = len(mat), len(mat[0])  
        # 直接用元组来存，(i,j)表示第i行中有j个1
        cc = [[0,0] for i in range(m)] 
        
        for i in range(m):
            cc[i][0] = i
            for j in range(n):
                if mat[i][j] == 1:
                    cc[i][1] += 1
                else: # 一行中1是连续的啊
                    break  
        
        cc.sort(key = lambda c:c[1]) # 数组按照第二项进行从小往大排序
        
        res = []
        for i in range(k):
            res.append(cc[i][0])
        
        return res

5329. Reduce Array Size to The Half

1. 题意

给定一个一维数组arr，你可以选择一个整数集合，然后把数组中所有出现在集合中的数都去掉，最后让数组的长度减半！求最短的集合

比如arr = [3,3,3,3,5,5,5,2,2,7]， set = {3,5} or {3,2} or {5,2} ， 删除set中在arr中出现的数，都可以导致arr的长度减半！因此最短的集合的长度为2！

2.思路

统计arr中每一个数出现的次数，然后从大到小排序，当然是从出现最多的数开始删除！

3. Python实现

class Solution(object):
    def minSetSize(self, arr):
        """
        :type arr: List[int]
        :rtype: int
        """
        n = len(arr)
        cc = collections.Counter(arr)
        c = sorted(cc.items(), key = lambda d:-d[1])  # 按照出现次数排序
        res = 0
        count = 0
        # print(c)
        for tup in c:
            if count < n//2:
                count += tup[1]
                res += 1
            else:
                break
            
        return res

5330. Maximum Product of Splitted Binary Tree

1. 题意

给定一棵二叉树root, 然后每次删除其中一条边，把二叉树分裂成两棵二叉树，求两棵二叉树的和的乘积的最大值！

2. 思路

我看到这道题，立马就思考去掉边怎么操作？还有就是子树如何求和？
数据量最大有50000，并且数值在0-10000之间，可以看出节点有重复值，并且时间复杂度不能是O(N^2)，否则就会超时！

所以最好就是在遍历的时候顺便求子树的和，并且顺便也求两个子树之间的乘积，这样时间复杂度就控制在O(N)了。

做这道题之前最好做一下另外一道题：
程序员面试算法宝典—解题总结: 第三章 二叉树 3.4 如何求一棵二叉树的最大子树和

会求二叉树的子树和就会做这道题了！

3. python实现

class Solution(object):
    def maxProduct(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        self.allsum = 0
        self.product = float('-inf')
        
        def dfs(root): # 求树的所有节点之和
            if not root: return 
            self.allsum += root.val
            dfs(root.left)
            dfs(root.right)
        
        def post(root): # 返回以root为根的子树的和,后序遍历，自底向上来遍历
            if not root: return 0
            lsums = post(root.left)
            rsums = post(root.right)
            subsum = lsums + rsums + root.val 
            # print(subsum, (self.allsum-subsum), subsum* (self.allsum-subsum))
            self.product = max(self.product, subsum* (self.allsum-subsum))
            return subsum
            
        dfs(root)
        post(root)
        return self.product % 1000000007

5331. Jump Game V