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 个正整数的排列,其中 n 是奇数,只有充分利用给定的条件,才能得到答案。
为了得到原始数组 perm,应首先得到数组 perm 的第一个元素(即下标为 0 的元素),这也是最容易得到的。如果能得到数组 perm 的全部元素的异或运算结果,以及数组 perm 除了 perm}[0] 以外的全部元素的异或运算结果,即可得到 perm}[0] 的值。
由于数组 perm 是前 n 个正整数的排列,因此数组 perm 的全部元素的异或运算结果即为从 1 到 n 的全部正整数的异或运算结果。用 total 表示数组 perm 的全部元素的异或运算结果,则有
\begin{aligned}
\textit{total} &= 1 \oplus 2 \oplus \ldots \oplus n \
&= \textit{perm}[0] \oplus \textit{perm}[1] \oplus \ldots \oplus \textit{perm}[n-1]
\end{aligned}
其中 \oplus 是异或运算符。
如何得到数组 perm 除了 perm}[0] 以外的全部元素的异或运算结果?由于 n 是奇数,除了 perm}[0] 以外,数组 perm 还有 n-1 个其他元素,n-1 是偶数,又由于数组 encoded 的每个元素都是数组 perm 的两个元素异或运算的结果,因此数组 encoded 中存在 n-1/2 个元素,这些元素的异或运算的结果为数组 perm 除了 perm}[0] 以外的全部元素的异或运算结果。
具体而言,数组 encoded 的所有下标为奇数的元素的异或运算结果即为数组 perm 除了 perm}[0] 以外的全部元素的异或运算结果。用 odd 表示数组 encoded 的所有下标为奇数的元素的异或运算结果,则有
\begin{aligned}
\textit{odd} &= \textit{encoded}[1] \oplus \textit{encoded}[3] \oplus \ldots \oplus \textit{encoded}[n-2] \
&= \textit{perm}[1] \oplus \textit{perm}[2] \oplus \ldots \oplus \textit{perm}[n]
\end{aligned}
根据 total 和 odd 的值,即可计算得到 perm}[0] 的值:
\begin{aligned}
\textit{perm}[0] &= (\textit{perm}[0] \oplus \ldots \oplus \textit{perm}[n]) \oplus (\textit{perm}[1] \oplus \ldots \oplus \textit{perm}[n]) \
&= \textit{total} \oplus \textit{odd}
\end{aligned}
当 1 \le i<n 时,有 encoded}[i-1]=\textit{perm}[i-1] \oplus \textit{perm}[i]。在等号两边同时异或 perm}[i-1],即可得到 perm}[i]=\textit{perm}[i-1] \oplus \textit{encoded}[i-1]。计算过程见「1720. 解码异或后的数组的官方题解 」。
由于 perm}[0] 已知,因此对 i 从 1 到 n-1 依次计算 perm}[i] 的值,即可得到原始数组 perm。
1 | class Solution { |
1 | public class Solution { |
1 | var decode = function(encoded) { |
1 | func decode(encoded []int) []int { |
1 | class Solution { |
1 | int* decode(int* encoded, int encodedSize, int* returnSize) { |
1 | class Solution: |
复杂度分析
时间复杂度:O(n),其中 n 是原始数组 perm 的长度。计算 total 和 odd 各需要遍历长度为 n-1 的数组 encoded 一次,计算原数组 perm 的每个元素值也需要遍历长度为 n-1 的数组 encoded 一次。
空间复杂度:O(1)。注意空间复杂度不考虑返回值。