给你一个整数数组 perm ,它是前 n 个正整数的排列,且 n 是个 奇数 。
它被加密成另一个长度为 n - 1 的整数数组 encoded ,满足 encoded[i] = perm[i] XOR perm[i + 1] 。比方说,如果 perm = [1,3,2] ,那么 encoded = [2,1] 。
给你 encoded 数组,请你返回原始数组 perm 。题目保证答案存在且唯一。
基本运算法则
- 归零律:a ⊕ a = 0
- 恒等律:a ⊕ 0 = a
- 交换律:a ⊕ b = b ⊕ a
- 结合律:a ⊕b ⊕ c = a ⊕ (b ⊕ c) = (a ⊕ b) ⊕ c;
- 自反:a ⊕ b ⊕ a = b.
基于这些基础知识再来看题目。
- 可以知道 perm 数组长度为 encoded 长度加一,且其中元素为 1 - n 数字。
- 有encoded[i] = perm[i] ⊕ perm[i + 1] => perm[i+1] = encoded[i] ⊕ perm[i]
根据例题 encoded = [6, 5, 4 ,6], perm = [2, 4, 1, 5, 3];
可以看到 6 ⊕ 4 = 2 ⊕ 4 ⊕ 1 ⊕ 5 ; 5 ⊕ 6 = 4 ⊕ 1 ⊕ 5 ⊕ 3 。
又 perm 由 1 - n 组成,故可以推出第一位和最后一位。
整理后思路如下:
- 求出 1 ⊕ 2 ⊕ … ⊕ n 的值
- 这里用 encoded[1] ⊕ … ⊕ encoded[-1] (这里用所有索引为奇数的值求出除第一位以外的异或,也可以用所有索引为偶数的求出最后一位)
- perm[i] = perm[i-1] ^ encoded[i-1] 求出后面值(如果步骤2求出了最后一位,就从后向前推)
代码如下:
class Solution {
public int[] decode(int[] encoded) {
int[] perm = new int[encoded.length+1];
int len = perm.length, n = 0;
//n = 1 ^ 2 ^ ... ^ n;
for(int i = 1 ; i <= len ; i++){
n ^= i;
}
//n = n ^ encoded[1] ^ ...encoded[2i+1]... ^ encoded[-1];
for(int i = 1 ; i < encoded.length ; i += 2){
n ^= encoded[i];
}
perm[0] = n;
//perm[i-1] ^ perm[i] = encoded[i-1] => perm[i] = perm[i-1] ^ encoded[i-1];
for(int i = 1 ; i < len ; i++){
perm[i] = perm[i-1] ^ encoded[i-1];
}
return perm;
}
}