leetcode 1734.解码异或后的排列

leetcode 1734.解码异或后的排列

题干

给你一个整数数组 perm ，它是前 n 个正整数的排列，且 n 是个 奇数 。
它被加密成另一个长度为 n - 1 的整数数组 encoded ，满足 encoded[i] = perm[i] XOR perm[i + 1] 。比方说，如果 perm = [1,3,2] ，那么 encoded = [2,1] 。
给你 encoded 数组，请你返回原始数组 perm 。题目保证答案存在且唯一。

示例 1：
输入：encoded = [3,1]
输出：[1,2,3]
解释：如果 perm = [1,2,3] ，那么 encoded = [1 XOR 2,2 XOR 3] = [3,1]

示例 2：
输入：encoded = [6,5,4,6]
输出：[2,4,1,5,3]

提示：
3 <= n < 105
n 是奇数。
encoded.length == n - 1

知识点&算法

整道题的关键有两个：一是条件中给出的，perm是前n个正整数的排列；二是异或同一个数两次还原的特性。
这里可以进行一个推导，用pi表示perm[i]，ei表示encoded[i]：

p0 xor p1 = e0
p1 xor p2 = e1
p2 xor p3 = e2
p3 xor p4 = e3
p0 = e0 xor p1
p1 = e1 xor p2
p2 = e2 xor p3
p3 = e3 xor p4

p0 xor e1 xor e3 = p0 xor p1 xor p2 xor p3 xor p4 = xorSum
p0 = xorSum xor e1 xor e3

恰好发现，当perm[0]和所有下标为奇数的encoded[i]异或后，得到的结果就是前n个正整数异或的结果(即上述的xorSum)。
因此我们只要先求出前n个正整数异或的结果，然后再异或上encoded中下标为奇数的元素，就能得到perm[0]。
得到perm[0]后不断和encoded异或，就能得到最终的序列。

题解

class Solution {
public:
    vector<int> decode(vector<int>& encoded) {
        int n = encoded.size() + 1, perm = 0;
        vector<int> res;
        for(int i = 1 ; i <= n ; ++i){
            perm ^= i;
            if(i != n && i % 2 == 1) perm ^= encoded[i];
        }
        res.push_back(perm);
        for(int i = 0 ; i < n - 1; ++i){
            perm ^= encoded[i];
            res.push_back(perm);
        }
        return res;
    }
};