[C++&Rust]LeetCode No.1734 解码异或后的排列(每日一题)

原帖地址：http://blog.leanote.com/post/dawnmagnet/fda5818930ff

题目

给你一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/decode-xored-permutation
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路分析

这是一道数学题，需要思考。
之前说过异或的最重要的性质就是两个相同的数字异或之后是0.所以我们一定要牢牢抓住这一点。
在encode数组中，每个项都是perm中的两项的异或，也就是说，我们只要知道了perm[0]，也就能通过encode[1]知道perm[1]，，也就能通过encode[2]知道perm[2]，也就。。。。。。。知道了所有。所以我们先写一段简单的，也就是通过perm[0]和encode求出最终的结果。

vector<int> res = {cur};
for (int i = 0; i < 2 * n; ++i) res.push_back(res[i] ^ encoded[i]);
return res;

我们再将工作的中心转移到求这个perm[0]上来，我们手里的工具只有encode数组，也就是说这个perm[0]必定可以通过encode求出，要不然题目就有错了。
我们先将encode中的所有项列出来看看规律。
假设perm中的数字有7个

encode中的下标	0	1	2	3	4	5	6
对应到perm中的下标	0+1	1+2	2+3	3+4	4+5	5+6	6+7

通过观察我们可以发现

• encode中所有项的异或是perm[0] ^ perm[7]
• encode中所有奇数项的和是 perm[1] ^ … ^ perm7
• encode中所有偶数项的和是 perm[0] ^ … ^ perm6

但其实我们是知道所有项的异或和的，也就是1²3⁴5⁶7，我们拿所有项的异或和与encode中奇数项的异或和再求一个异或，就得到了perm[0]

C++代码

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

Rust代码

impl Solution {
    pub fn decode(encoded: Vec<i32>) -> Vec<i32> {
        let n = encoded.len() / 2;
        let mut cur = (0..=(2 * n + 1)).fold(0, |acc, x| acc ^ x) as i32;
        for i in 0..n {
            cur ^= encoded[i * 2 + 1];
        }
        let mut res = vec![cur];
        for i in 0..(2 * n) {
            res.push(res[i] ^ encoded[i]);
        }
        res
    }
}