1. 程式人生 > 其它 >淺談 解碼異或後的陣列 問題

淺談 解碼異或後的陣列 問題

技術標籤:資料結構與演算法(刷題篇)leetcodec++

解碼異或後的陣列

問題:
未知 整數陣列 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; } };