思路:hash
难点:初始元素值各不相同->换成一个在nums[i]不相同的值->每轮操作后数组内的元素依旧不相同,因此每轮替换只会替换一个元素。
class Solution {
public:
vector<int> arrayChange(vector<int>& nums, vector<vector<int>>& operations) {
int n = nums.size();
unordered_map<int, int> hash;
for (int i = 0; i < n; ++i) {
hash[nums[i]] = i;
}
for (auto& each : operations) {
int a = each[0], b = each[1];
nums[hash[a]] = b;
hash[b] = hash[a];
}
return nums;
}
};
难度升级:如果每轮替换可能会修改多个下标呢?
此时就不能这么做了,O(n^2)的复杂度会超时
解决方案:倒着访问operations
3->6 5->6 6->7 也就是:3 5 6 都修改为7,先把6修改为7,产生映射6->7,修改5时,发现6存在映射,此时修改为7,产生5->hash[6]的映射,修改3时,再产生3->hash[5]的映射。
class Solution {
public:
vector<int> arrayChange(vector<int>& nums, vector<vector<int>>& operations) {
int n = nums.size(), m = operations.size();
unordered_map<int, int> hash;
for (int i = m - 1; i >= 0; --i) {
auto each = operations[i];
int a = each[0], b = each[1];
hash[a] = hash.count(b) ? hash[b] : b;
}
for (int i = 0; i < n; ++i) {
nums[i] = hash.count(nums[i]) ? hash[nums[i]] : nums[i];
}
return nums;
}
};
错误思路:并查集,3->6 5->6 6->7 7再回到3就出问题了,修改有时间的序列,不能用并查集修改。