LeetCode 1475. 商品折扣后的最终价格 / 687. 最长同值路径 / 652. 寻找重复的子树

1475. 商品折扣后的最终价格

2022.9.1 每日一题，转眼已经入职一个多月了

题目描述

给你一个数组 prices ，其中 prices[i] 是商店里第 i 件商品的价格。

商店里正在进行促销活动，如果你要买第 i 件商品，那么你可以得到与 prices[j] 相等的折扣，其中 j 是满足 j > i 且 prices[j] <= prices[i] 的 最小下标 ，如果没有满足条件的 j ，你将没有任何折扣。

请你返回一个数组，数组中第 i 个元素是折扣后你购买商品 i 最终需要支付的价格。

示例 1：

输入：prices = [8,4,6,2,3]
输出：[4,2,4,2,3]
解释：
商品 0 的价格为 price[0]=8 ，你将得到 prices[1]=4 的折扣，所以最终价格为 8 - 4 = 4 。
商品 1 的价格为 price[1]=4 ，你将得到 prices[3]=2 的折扣，所以最终价格为 4 - 2 = 2 。
商品 2 的价格为 price[2]=6 ，你将得到 prices[3]=2 的折扣，所以最终价格为 6 - 2 = 4 。
商品 3 和 4 都没有折扣。

示例 2：

输入：prices = [1,2,3,4,5]
输出：[1,2,3,4,5]
解释：在这个例子中，所有商品都没有折扣。

示例 3：

输入：prices = [10,1,1,6]
输出：[9,0,1,6]

提示：

1 <= prices.length <= 500
1 <= prices[i] <= 10^3

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/final-prices-with-a-special-discount-in-a-shop
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

class Solution:
    def finalPrices(self, prices: List[int]) -> List[int]:
        # 还是个单调栈
        # 遇到大的就放入，因为不能处理
        # 遇到小的就处理
        stack = []
        for i in range(len(prices)):
            while stack and prices[stack[-1]] >= prices[i]:
                top = stack.pop()
                prices[top] = prices[top] - prices[i]
            stack.append(i)
        return prices

1
2
3
4
5
6
7
8
9
10
11
12
13

687. 最长同值路径

2022.9.2 每日一题

题目描述

给定一个二叉树的 root ，返回 最长的路径的长度 ，这个路径中的 每个节点具有相同值 。 这条路径可以经过也可以不经过根节点。

两个节点之间的路径长度 由它们之间的边数表示。

示例 1:

输入：root = [5,4,5,1,1,5]
输出：2

示例 2:

输入：root = [1,4,5,4,4,5]
输出：2

提示:

树的节点数的范围是 [0, 10^4]
-1000 <= Node.val <= 1000
树的深度将不超过 1000

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/longest-univalue-path
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    reslen = 1
    def longestUnivaluePath(self, root: Optional[TreeNode]) -> int:
        # 最长同值路径，好像不太好做，想想该怎么做
        # 好像和之前做过的有个题差不都
        # 对于一个节点，左右有两条路径，可以加起来形成一个长的路径
        # 也可以继续往上延申，但是这时候的路径就是左右最长的路径加上当前节点往上的路径了
        # 所以先左后右再当前节点，就是后序遍历的思想
        # 记录左右边最长链条是多少，还有最大值是什么
        def postorder(root):
            if not root:
                return 0, -1001
            maxlen1, val1 = postorder(root.left)
            maxlen2, val2 = postorder(root.right)
            # 如果和两个值都相同，那么可以形成左中右的长链条
            if root.val == val1 and root.val == val2:
                maxlen = maxlen1 + maxlen2 + 1
                if maxlen > self.reslen:
                    self.reslen = maxlen
                # 返回的时候要返回左右的最长链条
                return (maxlen1 + 1, val1) if maxlen1 >= maxlen2 else (maxlen2 + 1, val1)
            # 如果左边连接上了，那么统计最长，返回左边
            elif root.val == val1:
                self.reslen = max(self.reslen, max(maxlen1 + 1, maxlen2))
                return maxlen1 + 1, val1
            elif root.val == val2:
                self.reslen = max(self.reslen, max(maxlen1, maxlen2 + 1))
                return maxlen2 + 1, val2
            # 如果都没有连接上，那么返回当前节点的值
            else:
                self.reslen = max(self.reslen, max(maxlen1, maxlen2))
                return 1, root.val
            
        postorder(root)
        return self.reslen - 1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

一直有个问题困扰了我一上午，就是这两个函数输出是一样的吗
然后我终于发现了，需要将返回的内容加括号，结果才是一样的。。。

def cal(maxlen1, maxlen2, val1):
    if maxlen1 >= maxlen2:
        return maxlen1 + 1, val1
    else:
        return maxlen2 + 1, val1
def cal2(maxlen1, maxlen2, val1):
    return maxlen1 + 1, val1 if maxlen1 >= maxlen2 else maxlen2 + 1, val1
1
2
3
4
5
6
7

因为左右返回的时候，是根据当前的长度的，所以没必要记录长度链条的值

class Solution:
    reslen = 1
    def longestUnivaluePath(self, root: Optional[TreeNode]) -> int:
        self.postorder(root)
        return self.reslen - 1

    def postorder(self, root):
        if not root:
            return 0, -1001
        ans = 1
        cur = 1
        left = self.postorder(root.left)
        right = self.postorder(root.right)
        # 如果和两个值都相同，那么可以形成左中右的长链条
        # 如果左边连接上了，那么统计最长，返回左边
        if root.left and root.val == root.left.val:
            ans = left + 1
            cur = cur + left
        if root.right and root.val == root.right.val:
            ans = max(ans, right + 1)
            cur = cur + right        
        self.reslen = max(self.reslen, cur)
        return ans
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

652. 寻找重复的子树

2022.9.5 每日一题

题目描述

给定一棵二叉树 root，返回所有重复的子树。

对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。

如果两棵树具有相同的结构和相同的结点值，则它们是重复的。

示例 1：

输入：root = [1,2,3,4,null,2,4,null,null,4]
输出：[[2,4],[4]]

示例 2：

输入：root = [2,1,1]
输出：[[1]]

示例 3：

输入：root = [2,2,2,3,null,3,null]
输出：[[2,3],[3]]

提示：

树中的结点数在[1,10^4]范围内。
-200 <= Node.val <= 200

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-duplicate-subtrees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    ll = []
    def findDuplicateSubtrees(self, root: Optional[TreeNode]) -> List[Optional[TreeNode]]:
        # 理解错了，是重复子树，而不是重复路径
        # 所以需要对子树进行序列化的操作
        memo = dict()
        res = []
        def encode(root):
            if not root:
                return " "
            s = str(root.val) + '_'
            s =  s + encode(root.left) + encode(root.right)
            memo[s] = memo.get(s, 0) + 1
            if memo[s] == 2:
                res.append(root)
            return s
        encode(root)
        return res
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

官解的这种将子树编号的方法，用（val，左子树编号，右子树编号）表示一个子树
如果这样的元组在哈希表中存在，那么就说明重复了
否则，加入哈希表中，并且对这棵树编号

class Solution:
    def findDuplicateSubtrees(self, root: Optional[TreeNode]) -> List[Optional[TreeNode]]:
        def dfs(node: Optional[TreeNode]) -> int:
            if not node:
                return 0
            
            tri = (node.val, dfs(node.left), dfs(node.right))
            # 如果这个编号已经存在，那么返回这个编号
            if tri in seen:
                (tree, index) = seen[tri]
                repeat.add(tree)
                return index
            # 否则，存储到哈希表中，将编号返回
            else:
                nonlocal idx
                idx += 1
                seen[tri] = (node, idx)
                return idx
        
        idx = 0
        seen = dict()
        repeat = set()

        dfs(root)
        return list(repeat)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25