题意:
链接:1787. 使所有区间的异或结果为零 - 力扣(LeetCode) (leetcode-cn.com)
给你一个整数数组 nums 和一个整数 k 。区间 [left, right](left <= right)的 异或结果 是对下标位于 left 和 right(包括 left 和 right )之间所有元素进行 XOR 运算的结果:nums[left] XOR nums[left+1] XOR ... XOR nums[right] 。返回数组中 要更改的最小元素数 ,以使所有长度为 k 的区间异或结果等于零。
解题思路:
首先对于每个区间大小为k的区间的异或为 0。可以得到:
nums[s]^nums[s+1]^...nums[s+k] = nums[s+1]^nums[s+2]^...nums[s+k+1]
所以:nums[s]^nums[s+1]^...nums[s+k] ^ nums[s+1]^nums[s+2]^...nums[s+k+1] = 0
所以:nums[s] ^ nums[s+k+1] = 0
所以:nums[s] = nums[s+k+1]
所以:对于nums数组来说,以k个为单位分割,每段相同;换句话说: a[0] = a[k], a[1] = a[k + 1], a[k - 1] = a[2*k - 1]。
所以:可以把nums数组看成 (nums.size() / k) * k 的矩阵。矩阵的相邻两行对应的元素,如果有,那么大小应该相等,且第一行的异或值为0;可视化就是:
当然也有可能不是正好的:
弄成二维的矩阵之后就会方便很多了,因为要修改数据肯定是要一列都改成一样的,原因见上面的 所以:所以:所以:所以:
所以可以用dp[i][j]来表示前 i 列异或为 j ,需要修改元素的个数。用mp保存每列中每个数出现的次数,统计好方便修改。
因为最大值为 2^10 - 1,所以 j 的取值范围是 0 - 2^10-1 ,设 第 i 列的元素的个数为 cnt
对于第0列: dp[0][j] = cnt - mp[j]; // 自己异或,就是自己, 需要修改的次数直接用一列中数的个数减去已经存在的数的个数。
对于第i列(i>0):
for(int j = 0; j < 1024; j++) for(int cur = 0; cur < 1024; cur++) dp[i][j] = min( dp[ i ][ j ] , dp[ i - 1 ][ j ^ cur ] + cnt - mp[cur] )
优化:
可以保存上一列修改次数最少的异或结果;假设上一列最少修改次数对应的异或结果值为 x,那么其实对于第 i 列,dp[ i ][ j ] = dp[i - 1][x] + cnt; // 相当于把第 i 列的所有元素改为了 j ^ x 。这样前 i 列的异或结果就为 j 了。
同时会发现并不需要指导具体的 x 是多少,只需要保存第 i - 1 列最小的次数就好,第 i - 1 列最小的的次数其实就是 dp[i - 1][x]。
但这样会浪费一些已有的值,所以现在在对已有的值进行下:for(int j = 0; j < 1024; j++) for(auto& t : mp) dp[i][j] = min( dp[ i ][ j ] , dp[ i - 1 ][ j ^ t.first ] + cnt - t.second );就OK了
AC代码:
class Solution {
public:
map<int, int> mp;
int minChanges(vector<int>& nums, int k) {
vector<vector<int> > dp(k + 1, vector<int>(1026, 0x3f3f3f3f));
vector<int> tmp(k, 0x3f3f3f3f); // 保存前i-1行最小修改次数
for(int i = 0; i < k; i++)
{
mp.clear();
int cnt = 0;
for(int t = i; t < nums.size(); t += k)
{
mp[nums[t]]++;
cnt++;
}
for(int j = 0; j < 1024; j++)
{
if(i == 0)
{
dp[0][j] = cnt - mp[j];
tmp[0] = min(tmp[0], dp[0][j]);
}
else
{
dp[i][j] = tmp[i - 1] + cnt;
for(auto& t : mp)
{
dp[i][j] = min(dp[i][j], dp[i - 1][j ^ t.first] + cnt - t.second);
}
tmp[i] = min(tmp[i], dp[i][j]);
}
}
}
return dp[k - 1][0];
}
};