给你一个长度为 n
的 整数 数组 pref
。找出并返回满足下述条件且长度为 n
的数组 arr
:
pref[i] = arr[0] ^ arr[1] ^ ... ^ arr[i]
.
注意 ^
表示 按位异或(bitwise-xor)运算。
可以证明答案是 唯一 的。
示例 1:
输入:pref = [5,2,0,3,1]
输出:[5,7,2,3,2]
解释:从数组 [5,7,2,3,2] 可以得到如下结果:
- pref[0] = 5
- pref[1] = 5 ^ 7 = 2
- pref[2] = 5 ^ 7 ^ 2 = 0
- pref[3] = 5 ^ 7 ^ 2 ^ 3 = 3
- pref[4] = 5 ^ 7 ^ 2 ^ 3 ^ 2 = 1
示例 2:
输入:pref = [13]
输出:[13]
解释:pref[0] = arr[0] = 13
提示:
1 <= pref.length <= 10^5
0 <= pref[i] <= 10^6
解法 异或前缀和的还原
问题本质上是给你前缀和,要求还原回原数组。由于前缀和的差分是原数组,因此直接两两异或即可。
class Solution {
public:
vector<int> findArray(vector<int>& pref) {
int n = pref.size();
vector<int> ans(n);
ans[0] = pref[0];
for (int i = 1; i < n; ++i) ans[i] = pref[i] ^ pref[i - 1];
return ans;
}
};