题意:
给你一个整数数组 nums 和一个整数 k 。
区间 [left, right](left <= right)的 异或结果,
是对下标位于 left 和 right(包括 left 和 right )之间所有元素进行 XOR 运算的结果:
nums[left] XOR nums[left+1] XOR ... XOR nums[right] 。
返回数组中 要更改的最小元素数 ,以使所有长度为 k 的区间异或结果等于零。
数据范围:
1 <= k <= nums.length <= 2000
0 <= nums[i] < 2^10
解法:
设下标为[0,n-1]
因为所有长度为k的区间异或和为0,
那么最后的序列一定是[0,k-1]序列的循环,
我们只需要考虑[0,k-1]序列是什么样的就行了.
将a[i]按%k分组,那么每一组的所有数需要改成相同的数.
令d[i][j]表示前i组,异或和为j的最小修改次数,
cnt[i][j]表示第i组,数字j的个数,
num[i]表示第i组的数字总数量.
1.如果第i组修改的值是第i组中原有的数,那么有一部分位置是不用修改的.
枚举第i组中原有的数p,
那么d[i][j]=min(d[i][j],d[i-1][p^j]+(num[i]-cnt[i][p])).
2.如果第i组修改的值是第i组中没有的数,那么所有位置都需要修改,
但是一个好处是因为可以任选修改的数,所以可以组成任意异或和,
由于不管前面的数是多少都可以组成任意异或和,因此d[i-1]直接取最小值即可,
d[i][j]=min(d[i][j],min{d[i-1]}+num[i]).
code:
const int maxm=2e3+5;
vector<int>g[maxm];
int cnt[maxm][maxm];
int d[maxm][maxm];
int num[maxm];
class Solution {
public:
int minChanges(vector<int>& a, int k) {
memset(d,0,sizeof d);
memset(cnt,0,sizeof cnt);
memset(num,0,sizeof num);
for(int i=0;i<maxm;i++)g[i].clear();
int n=a.size();
for(int i=0;i<n;i++){
cnt[i%k][a[i]]++;
num[i%k]++;
g[i%k].push_back(a[i]);
}
for(int j=0;j<(1<<10);j++){
d[0][j]=1e9;
}
d[0][0]=0;
for(int i=1;i<=k;i++){
int mi=1e9;
for(int j=0;j<(1<<10);j++)mi=min(mi,d[i-1][j]);
for(int j=0;j<(1<<10);j++){
d[i][j]=1e9;
for(auto p:g[i-1]){
d[i][j]=min(d[i][j],d[i-1][p^j]+(num[i-1]-cnt[i-1][p]));
}
d[i][j]=min(d[i][j],mi+num[i-1]);
}
}
return d[k][0];
}
};