【leetCode】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 < 10^5
• n 是奇数。
• encoded.length == n - 1

解题思路

从数学角度上来讲，满足递推关系的序列求解，一般是从序列的首尾入手。

先来寻找几个规律：

1.数组perm是前n个正整数的排列，且n是奇数。

n = 1 : perm[0] = 1;

n = 3 : perm[0] ^ perm[1] ^ perm[2] = 0;

n = 5 : perm[0] ^ perm[1] ^ perm[2] ^ perm[3] ^ perm[4]= 1;

n = 7 : perm[0] ^ perm[1] ^ perm[2] ^ perm[3] ^ perm[4] ^ perm[5] ^ perm[6]= 0;

n = 9 : perm[0] ^ perm[1] ^ perm[2] ^ perm[3] ^ perm[4] ^ perm[5] ^ perm[6] ^ perm[7]= 1;

……

一般结论：当 n % 4 = 1时，perm[0] ^ perm[1] ^……^perm[n - 1] = 1;

                  当 n % 4 = 3时，perm[0] ^ perm[1] ^……^perm[n - 1] = 0;

2.encoded[i] = perm[i] XOR perm[i + 1].

encoded[0] = perm[0] ^ perm[1];

encoded[1] = perm[1] ^ perm[2];

encoded[2] = perm[2] ^ perm[3];

encoded[3] = perm[3] ^ perm[4];

encoded[4] = perm[4] ^ perm[5];

……

一般结论：可以通过encoded数组灵活组合求perm序列的异或值。

比如：perm[0] ^ perm[n - 1] = encoded[0] ^ encoded[1] ^ encoded[2] ^ …… ^ encoded[n - 2];

          perm[0] ^ perm[1] ^ perm[2] ^ ……^ perm[n - 2]  = encoded[0] ^ encoded[2] ^ encoded[4] ^ …… ^ encoded[2*i](i为偶数)；

          perm[1] ^ perm[2] ^ perm[3] ^ ……^ perm[n - 1]  = encoded[1] ^ encoded[3] ^ encoded[5] ^ …… ^ encoded[2*i + 1](i为奇数)；

         ……

开始解题

1.  先解出perm[0]：

利用：

1. 当 n % 4 = 1时，perm[0] ^ perm[1] ^……^perm[n - 1] = 1;当 n % 4 = 3时，perm[0] ^ perm[1] ^……^perm[n - 1] = 0;
2.  perm[1] ^ perm[2] ^ perm[3] ^ ……^ perm[n - 1]  = encoded[1] ^ encoded[3] ^ encoded[5] ^ …… ^ encoded[2*i + 1](i为奇数)；

解出perm[0]的值；

再利用题目条件encoded[i] = perm[i] XOR perm[i + 1]，逐个解出perm[i].

     2. 先解出perm[n - 1]：

利用：

1. 当 n % 4 = 1时，perm[0] ^ perm[1] ^……^perm[n - 1] = 1;当 n % 4 = 3时，perm[0] ^ perm[1] ^……^perm[n - 1] = 0;
2. perm[0] ^ perm[1] ^ perm[2] ^ ……^ perm[n - 2] =   encoded[0] ^ encoded[2] ^ encoded[4] ^ …… ^ encoded[2*i](i为偶数)；

解出perm[n - 1]的值；

再利用题目条件encoded[i] = perm[i] XOR perm[i + 1]，逐个解出perm[i].

代码实现

解perm[0]:

//计算perm[0]的值
vector<int> decode1(vector<int>& encoded) {
	int n = encoded.size();
	vector<int> perm(n + 1);
	int odd = 0;
	for (int i = 1; i < n; i += 2) {
		odd ^= encoded[i];
	}
	if ((n + 1) % 4 == 1) {
		perm[0] = 1 ^ odd;
	} else {
		perm[0] = odd;
	}

	for (int i = 1; i < n + 1; i++) {
		perm[i] = encoded[i - 1] ^ perm[i - 1];
	}
	return perm;
}

解perm[n - 1]:

这里用了一个更麻烦的推导，但是基本原理是一样的。

//计算perm[n - 1]的值
vector<int> decode(vector<int>& encoded) {
	int n = encoded.size();
	vector<int> perm(n + 1);
	//tmpXor[i] = prem[i] ^ prem[n]
	vector<int> tmpXor(n);
	tmpXor[n - 1] = encoded[n - 1];
	for (int i = n - 2; i >= 0; i--) {
		tmpXor[i] = encoded[i] ^ tmpXor[i + 1];
	}
	int sXorE = 0;
	//计算perm[0]^perm[1]^···^perm[n-2]的值
	for (int i = 0; i < n; i++) {
		sXorE ^= tmpXor[i];
	}
	//如果n % 4 == 1,那么perm[0]^perm[1]^···^perm[n-1] == 1
	//否则perm[0]^perm[1]^···^perm[n-1] == 0
	if ((n + 1) % 4 == 1) {
		perm[n] = 1 ^ sXorE;
	} else {
		perm[n] = sXorE;
	}
	//逐个计算perm的值
	for (int i = n - 1; i >= 0; i--) {
		perm[i] = encoded[i] ^ perm[i + 1];
	}

	return perm;
}