解码异或后的数组
问题:
未知 整数数组 arr 由 n 个非负整数组成。
经编码后变为长度为 n - 1 的另一个整数数组 encoded ,其中encoded[i] = arr[i] XOR arr[i + 1]
。例如,arr = [1,0,2,1] 经编码后得到 encoded = [1,2,3] 。
给你编码后的数组 encoded 和原数组 arr 的第一个元素 first(arr[0])。
请解码返回原数组 arr 。可以证明答案存在并且是唯一的。
思路:
例:
输入:encoded = [1,2,3], first = 1
输出:[1,0,2,1]
解释:若 arr = [1,0,2,1] ,那么 first = 1 且 encoded = [1 XOR 0, 0 XOR 2, 2 XOR 1] = [1,2,3]
再由 a ^ b = c → a = b ^ c 便可以写出代码段
class Solution {
public:
vector<int> decode(const vector<int>& encoded, const int first) {
vector<int> arr(encoded.size() + 1);
arr[0] = first;
for(int i = 0; i < encoded.size(); ++i){
arr[i +1] = arr[i] ^ encoded[i]; // a ^ b = c → a = b ^ c
}
return arr;
}
};