给你一个整数数组 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
思路:一道比较有意思的位运算题,下面说下具体思路,由于原数组是1到n的一个排列,假设原数组 arr为 [x,y,z,h,k],题目中的encode数组为[a,b,c,d]
则:a=x^y, b=y^z, c=z^h d=h^k;
我们首先预处理出来原数组所有元素的异或和:x^y^z^h^k,基于异或的性质我们知道只要能知道原数组的一个数,我们就能确定出最后整个数组。
我们观察下encode数组,我们发现,b^d=y^z^h^k,此时发现刚好少了一个a,之后让原数组异或和与这个数相异或不刚好得到a了嘛,问题解决了!
class Solution {
public int[] decode(int[] encoded) {
int n = encoded.length + 1;
int[] ans = new int[n];
int start = 0;
for (int i = 1; i <= n; i++) start ^= i;
for (int i = 1; i < n - 1; i += 2) start ^= encoded[i];
ans[0] = start;
for (int i = 0; i < n - 1; i++)
ans[i + 1] = ans[i] ^ encoded[i];
return ans;
}
}