LeetCode笔记：Biweekly Contest 103

原创于 2023-04-30 18:42:06 发布 · 762 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#双周赛 103 #leetcode 2656 #leetcode 2657 #leetcode 2658 #leetcode 2659

leetcode笔记专栏收录该内容

671 篇文章

订阅专栏

本文是关于LeetCode双周赛103的解题笔记，详细介绍了四个问题的解题思路和Python代码实现，包括最大和问题、寻找两个数组的公共前缀数组、网格中最多鱼的计算，以及最后一道使用FenwickTree或BinaryIndexedTree解决的题目。

LeetCode笔记：Biweekly Contest 103

比赛链接：https://leetcode.com/contest/biweekly-contest-103

1. 题目一

给出题目一的试题链接如下：

2656. Maximum Sum With Exactly K Elements

1. 解题思路

这一题描述得有点复杂，但是其实就是取最大值算一个等差数列求和即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def maximizeSum(self, nums: List[int], k: int) -> int:
        x = max(nums)
        return (x + x+k-1)  * k // 2

提交代码评测得到：耗时198ms，占用内存16.3MB。

2. 题目二

给出题目二的试题链接如下：

2657. Find the Prefix Common Array of Two Arrays

1. 解题思路

这一题要看某一个位置上两侧都有多少个元素已经出现过了，我们事实上只要分别找到两个数组当中某个元素最早同时出现的位置即可，之后后续所有的位置上求总数都得带有这个值，因此就变成了一个累计数组的问题了。

2. 代码实现

给出python代码实现如下：

class Solution:
    def findThePrefixCommonArray(self, A: List[int], B: List[int]) -> List[int]:
        n = len(A)
        cnt = [0 for _ in range(n)]
        index = {}
        for i, x in enumerate(A):
            index[x] = i
        for i, x in enumerate(B):
            cnt[max(i, index[x])] += 1
        return list(accumulate(cnt))

提交代码评测得到：耗时151ms，占用内存16.4MB。

3. 题目三

给出题目三的试题链接如下：

2658. Maximum Number of Fish in a Grid

1. 解题思路

这一题有多种思路，我这边采用了一个比较直接的思路，就是把每一个池子用DSU聚类，然后对每一个池子里的鱼求和然后返回最大值。

2. 代码实现

给出python代码实现如下：

class DSU:
    def __init__(self, n, m):
        self.width = m
        self.root = [i for i in range(n * m)]
        
    def _find(self, k):
        if self.root[k] == k:
            return k
        return self._find(self.root[k])
        
    def find(self, i, j):
        k = i * self.width + j
        if self.root[k] == k:
            return k
        return self._find(self.root[k])
    
    def union(self, i, j, a, b):
        x = self.find(i, j)
        y = self.find(a, b)
        if x != y:
            self.root[y] = x
        return

class Solution:
    def findMaxFish(self, grid: List[List[int]]) -> int:
        n, m = len(grid), len(grid[0])
        
        dsu = DSU(n, m)
        
        def is_water(i, j):
            return 0 <= i < n and 0 <= j < m and grid[i][j] != 0
        
        for i in range(n):
            for j in range(m):
                if grid[i][j] == 0:
                    continue
                if is_water(i+1, j):
                    dsu.union(i, j, i+1, j)
                if is_water(i, j+1):
                    dsu.union(i, j, i, j+1)
        
        cnt = defaultdict(int)
        for i in range(n):
            for j in range(m):
                cnt[dsu.find(i, j)] += grid[i][j]
        return max(cnt.values())