leetcode2766. 重新放置石块

---

官解

class Solution:
    def relocateMarbles(self, nums: List[int], moveFrom: List[int], moveTo: List[int]) -> List[int]:
        mp = {}  # 用于记录每个数字是否存在
        ans = []  # 存储最终结果的列表
        
        # 初始化 mp，将 nums 列表中的所有数字记录在 mp 中
        for num in nums:
            mp[num] = True
        
        # 遍历 moveFrom 和 moveTo 列表，根据移动操作更新 mp
        for i in range(len(moveFrom)):
            if moveFrom[i] in mp:
                del mp[moveFrom[i]]  # 删除 moveFrom 中指定的位置的数字
            mp[moveTo[i]] = True  # 在 moveTo 中指定的位置添加数字
        
        # 将 mp 的键转换为列表并排序
        ans = list(mp.keys())
        ans.sort()
        
        return ans  # 返回排序后的结果列表

---

chatgpt

代码分解与解释

变量和数据结构

• nums: List[int]：初始的数字列表。
• moveFrom: List[int] 和 moveTo: List[int]：分别表示从哪个位置移动和移动到哪个位置的列表。
• mp: dict：用来记录每个数字是否存在的字典。
• ans: List[int]：存储最终结果的列表。

算法步骤

1. 初始化字典 mp：

   使用字典 mp 记录 nums 列表中的每个数字是否存在。

2. 根据 moveFrom 和 moveTo 列表更新字典 mp：

   遍历 moveFrom 和 moveTo 列表，如果 moveFrom[i] 中的数字存在于 mp 中，则删除该数字，并在 moveTo[i] 中添加新数字。

3. 生成并排序结果列表 ans：

   将 mp 的键转换为列表并排序。

返回结果

• 返回排序后的结果列表 ans。

代码的关键点

1. 字典的使用：

   使用字典 mp 记录和更新每个数字是否存在，方便高效地进行查找和删除操作。

2. 遍历和更新：

   通过遍历 moveFrom 和 moveTo 列表，更新字典 mp 中的数字，确保每次移动操作都被正确处理。

3. 排序结果：

   最终将字典 mp 的键转换为列表并排序，确保返回的结果按升序排列。

总结

这段代码通过以下步骤实现了对 nums 列表中数字的重新定位：

1. 初始化字典 mp，记录 nums 列表中的每个数字。
2. 遍历 moveFrom 和 moveTo 列表，根据指定的移动操作更新字典 mp。
3. 将字典 mp 的键转换为列表并排序，得到最终的结果列表。

通过字典的高效查找和删除操作，该方法能够快速地处理大量的移动操作，并保证结果的正确性和有序性。