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

题意:

链接: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];

    }
};

  

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值