每日一解 找出变位映射

题目 找出变位映射

给定两个列表 Aand B，并且 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] 的整数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-anagram-mappings

思路

显然用哈希表存储一个表的数据，这样只要遍历两个表的元素一次就可以得到最终答案。如果要优化的话，大概就是可以利用长度为100这个特点，自己设计一个将数据映射到1——100范围内的哈希函数。
其中对于等值元素的情况，我采用向量存储，每次返回末尾的值，并删掉末尾，这样速度可以快一些，就是空间复杂度肯定会高一些了。

代码实现

class Solution {
public:
	vector<int> anagramMappings(vector<int>& A, vector<int>& B) {
		unordered_map<int, vector<int>> data;
		for (int i = 0; i < B.size(); i++) {
			data[B[i]].push_back(i);
		}
		vector<int> answer;
		for (int i = 0; i < A.size(); i++) {
			answer.push_back(data[A[i]][data[A[i]].size() - 1]);
			data[A[i]].pop_back();
		}
		return answer;
	}
};