LeetCode1734：解码异或后的排列

目录

一、题目

二、示例

三、思路

四、代码

---

一、题目

给你一个整数数组 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、题目：encoded[i] = perm[i] XOR perm[i + 1]，因此可得

                encoded[0] = perm[0] XOR perm[1]

                encoded[1] = perm[1] XOR perm[2]

                encoded[2] = perm[2] XOR perm[3]

                ...

                encoded[n - 1] = perm[n - 2] XOR perm[n - 1]

                将奇数项进行异或运算，得到：odd = encoded[1] XOR encoded[3] XOR ... XOR encoded[n - 2] = perm[1] XOR perm[2] XOR ... XOR[n - 1]

2、题目：整数数组 perm ，它是前 n 个正整数的排列，且 n 是个 奇数，因此得到：

                total = perm[0] XOR perm[1] XOR perm[2] XOR ... XOR perm[n - 1] = 1 XOR 2 XOR 3 XOR ... XOR n

3、根据 total 和 odd 可知：

      total XOR odd = perm[0]

4、得到perm[0]后，则该题和 1720.解码异或后的数组 相同。1720的题解可点击链接 LeetCode1720：解码异或后的数组！

四、代码

from functools import reduce
class Solution:
    def decode(self, encoded):
        """
        :param encoded: List[int]
        :return: List[int]
        """
        n = len(encoded) + 1
        total = reduce(lambda x, y: x ^ y, range(1, n + 1))

        odd = 0
        for i in range(1, n - 1, 2):
            odd = odd ^ encoded[i]

        perm = [total ^ odd]

        for i in range(n - 1):
            perm.append(perm[-1] ^ encoded[i])
        return perm

if __name__ == '__main__':
    # encoded = [3, 1]
    encoded = [6, 5, 4, 6]
    s = Solution()
    ans = s.decode(encoded)
    print(ans)