Leetcode P1787. 使所有区间的异或结果为零

Leetcode P1787. 使所有区间的异或结果为零

给你一个整数数组 nums​​​ 和一个整数 k​​​​​ 。区间 [left, right]（left <= right）的 异或结果 是对下标位于 left 和 right（包括 left 和 right ）之间所有元素进行 XOR 运算的结果：nums[left] XOR nums[left+1] XOR … XOR nums[right] 。返回数组中 要更改的最小元素数 ，以使所有长度为 k 的区间异或结果等于零。

示例 1:

输入：nums = [1,2,0,3,0], k = 1
输出：3
解释：将数组 [1,2,0,3,0] 修改为 [0,0,0,0,0]

示例 2:

输入：nums = [3,4,5,2,1,7,3,4,7], k = 3
输出：3
解释：将数组 [3,4,5,2,1,7,3,4,7] 修改为 [3,4,7,3,4,7,3,4,7]

示例 3:

输入：nums = [1,2,4,1,2,5,1,2,6], k = 3
输出：3
解释：将数组[1,2,4,1,2,5,1,2,6] 修改为 [1,2,3,1,2,3,1,2,3]

提示：

• 1 <= k <= nums.length <= 2000
• ​​​​​​0 <= nums[i] < 2^10

一开始看到没什么思路。本题的关键在于，由：
$$a_{i}XOR{a}_{i+1}XOR\cdots a_{i+k-1}=0$$
$$a_{i+1}XOR{a}_{i+2}XOR\cdots a_{i+k}=0$$

可以推出，$a_{i}XOR{a}_{i+k}=0$，即$a_i=a_{i+k}$。所以一旦我们定下前k个元素，之后的所有元素也都确定了。所以我们只需讨论前k个元素的取法情况，求出每种取法需要修改的次数。

可以按模k的不同把数组分成k组，每组都以第一个元素为代表，讨论这个第一个元素的取值情况。其实我们看到元素的最大值不超过1024，就可以想到由此入手，而且考虑到k个数异或值为0，k-1个数的异或可以有很多值。所以用dp[j]作为一个滚动数组，表示在当前第i组异或值为j的最小修改次数。

• mp记录第i组数字的出现次数，sz是第i组的大小
• 初始化下一次的dp数组为tmp
• 用mp做滚动更新

class Solution {
public:
    int minChanges(vector<int>& nums, int k) {
        unordered_map<int, int> mp[k];
        int sz[k];
        int n = nums.size();
        for (int i = 0; i < k; i++) {
            mp[i].clear(), sz[i] = 0;
            for (int j = i; j < n; j += k) {
                mp[i][nums[j]]++;
                sz[i]++;
            }
        }
        int dp[1024];
        for (int i = 1; i < 1024; i++) dp[i] = 1e9;
        dp[0] = 0;
        for (int i = 0; i < k; i++) {
            int tmp[1024];
            int minv = *min_element(dp, dp + 1024);
            for (int j = 0; j < 1024; j++) tmp[j] = minv + sz[i];
            for (int j = 0; j < 1024; j++) {
                if (dp[j] == 1e9) continue;
                for (auto it : mp[i]) {
                    int key = it.first, cnt = it.second;
                    tmp[j ^ key] = min(tmp[j ^ key], dp[j] + sz[i] - cnt);
                }
            }
            memcpy(dp, tmp, sizeof(tmp));
        }
        return dp[0];
    }
};