LeetCode-Python-760. 找出变位映射

最新推荐文章于 2023-08-08 16:30:04 发布

暴躁老哥在线刷题

最新推荐文章于 2023-08-08 16:30:04 发布

阅读量726

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/qq_32424059/article/details/90372562

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

博客围绕找出两个变位列表A和B的索引映射展开，列表可能有重复元素，若有多种答案输出任意一种。给出示例说明映射关系，还提到A和B长度范围及元素范围。思路是用hashmap记录B中元素和下标，再扫描A获取对应下标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定两个列表 Aand B，并且 B 是 A 的变位。B 是 A 的变位的意思是 B 由 A 中的元素随机排列生成。

我们希望找出一个从 A 到 B 的索引映射 P 。一个映射 P[i] = j 的意思是 A 中的第 i 个元素出现于 B 中的第 j 个元素上。

列表 A 和 B 可能出现重复元素。如果有多于一种答案，输出任意一种。

例如，给定

A = [12, 28, 46, 32, 50]
B = [50, 12, 32, 46, 28]

需要返回

[1, 4, 3, 2, 0]

P[0] = 1 ，因为 A 中的第 0 个元素出现于 B[1]，而且 P[1] = 4 因为 A 中第 1个元素出现于 B[4]，以此类推。

注：

A, B 有相同的长度，范围为 [1, 100]。
A[i], B[i] 都是范围在 [0, 10^5] 的整数。

思路：

用hashmap记录B中元素和对应下标，再线性扫描A，从hashmap里读每个元素在B里的下标就好了。

class Solution(object):
    def anagramMappings(self, A, B):
        """
        :type A: List[int]
        :type B: List[int]
        :rtype: List[int]
        """
        record = dict()
        for i, b in enumerate(B):
            record[b] = i
            
        res = [-1 for _ in range(len(A))]
        for i, a in enumerate(A):
            res[i] = record[a]
            
        return res