LeetCode 5700. 使所有区间的异或结果为零(dp)

题意:
给你一个整数数组 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) {
        //init
        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]){//因为只有n个数,所以这部分的总复杂度是O(n)的
                    d[i][j]=min(d[i][j],d[i-1][p^j]+(num[i-1]-cnt[i-1][p]));
                }
                //不用原有的数,可以和组成任意异或和,因此d[i-1]选择代价最小的即可.
                d[i][j]=min(d[i][j],mi+num[i-1]);
            }
        }
        return d[k][0];
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值