淺談 解碼異或後的陣列 問題
阿新 • • 發佈:2021-01-17
解碼異或後的陣列
問題:
未知 整數陣列 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;
}
};