力扣1787 使所有区间的异或结果为零

题目描述

     ~~~~     给你一个整数数组 nums​​​ 和一个整数 k​​​​​ 。区间 [left, right]（left <= right）的 异或结果 是对下标位于 left 和 right（包括 left 和 right ）之间所有元素进行 XOR 运算的结果：nums[left] XOR nums[left+1] XOR … XOR nums[right] 。

     ~~~~     返回数组中 要更改的最小元素数 ，以使所有长度为 k 的区间异或结果等于零。

解题思路

Hints:

1. nums[i] = nums[i + k]
2. 先确定前k个数如何处理

     ~~~~     根据提示，这个数组在修改后具有很强的周期性，因此可以分成k组处理，这k组的元素在处理过后应该相同，最终需要求出更改的最小元素数也就是保留的最大元素数，因此可分为以下两种情况处理：
     ~~~~      1.这k组最后选取的元素都是曾经出现过的，用dp[$i$][$j$]表示处理到第i组时,异或值为j保留的最大元素个数,得到转移方程:
     ~~~~      dp[0][$i$] = cnt[0][$i$] //cnt[$i$][$j$] 表示第$i$组元素$j$出现的次数
     ~~~~      dp[$i$][$j$] = max(dp[$i$][$j$],dp[$i$-1][$k$ ^ $j$] + cnt[$i$][$k$]) //其中cnt[$i$][$k$] > 0
     ~~~~      为了快速得到满足cnt[$i$][$k$] > 0 的k值，可以用map来存储cnt数组
     ~~~~      2. 有k-1组元素最终选取的元素是曾经出现过的，最后一组元素不曾出现，满足k个数异或为0即可。

代码

int minChanges(vector<int>& nums, int k) {
        vector<unordered_map<int,int> >cnt = vector<unordered_map<int,int> >(k);
        for(int i = 0;i < nums.size();i++)
            cnt[i%k][nums[i]]++; //O(n)
        int min = 0x7fffffff;
        int now_num = 0;
        for(int i = 0;i < k;i++)
        {
            int max_cnt = 0;
            for(auto v:cnt[i])
                if(v.second > max_cnt)
                    max_cnt = v.second;
            now_num += max_cnt;
            if(max_cnt < min)
                min = max_cnt;
        }//O(n/k * k) = O(n)

        int fin_keep = now_num - min;
        
        int dp[k][2048];
        // dp[i][j]; 处理到第i组时,异或值为j保留的最大元素个数
        memset(dp,0,sizeof(dp));
        for(int i = 0;i < 1024;i++)
            dp[0][i] = cnt[0][i];
        //dp[0][i] = cnt[0][i]
        //dp[i][j] = max(dp[i][j],dp[i-1][k ^ j] + cnt[i][k])
        for(int i = 1;i < k;i++)
            for(auto v: cnt[i])
                for(int j = 0;j < 2048;j++)
                    dp[i][j] = max(dp[i][j],v.second + dp[i-1][v.first ^ j]); //O(k*n/k*2048) = O(n)
        
        return nums.size() - max(dp[k-1][0],fin_keep);
    }