原帖地址: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
但其实我们是知道所有项的异或和的,也就是1234567,我们拿所有项的异或和与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
}
}