高效变位映射

原创

东百牧码人 2025-01-07 15:38:09 ©著作权

©著作权归作者所有：来自51CTO博客作者东百牧码人的原创作品，请联系作者获取转载授权，否则将追究法律责任

技术文章：解决变位映射问题的高效方法

引言

在编程领域，处理数组和字符串的变位词问题是一个常见的挑战。变位词是指通过重新排列元素顺序而形成的数组或字符串。例如，数组 [12, 28, 46, 32, 50] 和 [50, 12, 32, 46, 28] 是彼此的变位词。本文将介绍如何高效地解决一个特定的变位映射问题：给定两个整数数组 nums1 和 nums2，其中 nums2 是 nums1 的一个变位词，我们需要找到一个映射数组 mapping，使得 mapping[i] = j 表示 nums1 中的第 i 个元素出现在 nums2 的第 j 个下标上。

问题描述

输入：两个整数数组 nums1 和 nums2，其中 nums2 是 nums1 的一个变位词。
输出：一个下标映射数组 mapping，满足 mapping[i] = j，表示 nums1[i] 在 nums2 中的下标为 j。
约束：如果有多个答案，返回任意一个即可。

解题思路

要解决这个问题，我们可以利用哈希表（HashMap）来存储 nums2 中每个元素的下标信息。具体步骤如下：

构建哈希表：遍历 nums2，将每个元素及其对应的下标存入哈希表中。这样，我们可以在常数时间内查找 nums2 中任意元素的下标。
生成映射数组：遍历 nums1，对于每个元素，通过哈希表查找其在 nums2 中的下标，并将结果存入映射数组 mapping 中。

代码实现

以下是使用 Java 实现的代码示例：

import java.util.HashMap;
import java.util.Map;

public class LeetCode760 {
    public int[] anagramMappings(int[] A, int[] B) {
        // 创建一个哈希表来存储 nums2 中每个元素的下标
        Map<Integer, Integer> indexMap = new HashMap<>();
        for (int i = 0; i < B.length; ++i) {
            indexMap.put(B[i], i);
        }

        // 创建映射数组
        int[] mapping = new int[A.length];
        for (int i = 0; i < A.length; ++i) {
            // 通过哈希表查找 nums1 中元素在 nums2 中的下标
            mapping[i] = indexMap.get(A[i]);
        }

        return mapping;
    }
}