力扣 1787. 使所有区间的异或结果为零 dp 位运算

https://leetcode-cn.com/problems/make-the-xor-of-all-segments-equal-to-zero/

思路：依据题意，有：
$$\begin{gathered} nums[i]xornums[i+1]\dots \dots xornums[i+k-1]=0(1) \\ nums[i+1]xornums[i+2]\dots \dots xornums[i+k]=0(2) \end{gathered}$$
$(1)、(2)$异或可得：$nums[i]xornums[i+k]=0$，即$nums[i]=nums[i+k]$，因此最后符合题意的数组一定是有循环节的，循环长度为$k$。因此我们可以把数组分为$k$组，考虑用$dp[i][j]$表示处理到第$i$组且$[0,i]$组所有元素异或和($nums[0]xor\dots \dots nums[i]$)为$j$的情况下所需要更改的最小元素数。

对于第$i$组元素，假设把它们全部变化为$x$，那么有：
$$dp[i][j]=min(dp[i-1][jxorx]+size(i)-count(x))$$
其中$size(i)$为第$i$组元素的个数，$count(x)$为元素$x$在该组内出现的次数。考虑到$nums[i]<2^{10}$，因此上述算法的复杂度为$O(k\ast 2^{10}\ast 2^{10})$，会超时。

接下来考虑优化，首先$size(i)$与$x$无关，可以提出去；然后考虑当$count(x)=0$时，上述公式可以转换为：
$$dp[i][j]=min(dp[i-1][jxorx])+size(i)(3)$$
对于确定的$i、j$，这个可以在$O(2^{10})$内计算出来。

当$count(x)\ne 0$时，我们依然需要枚举$x$，但是可以只枚举出现过的$x$，这时公式依然为：
$$dp[i][j]=min(dp[i-1][jxorx]-count(x))+size(i)(4)$$
但是我们可以发现当$count(x)\ne 0$时，$(4)$一定要小于$(3)$，由于我们取的是最小值，因此把$(4)$一起考虑也没有关系。所以不管$x$是否在该组出现过，我们都可以先令$dp[i][j]=min(dp[i-1][jxorx])+size(i)$，然后再枚举出现过的$x$用$(4)$更新。这样复杂度可以降低到$O((k+n)\ast 2^{10})$。

class Solution {
public:
    int minChanges(vector<int>& nums, int k) {
        const int maxn=1<<10;
        const int inf=0x3f3f3f3f;
        int n=nums.size();
        vector<int> dp(maxn,inf);
        dp[0]=0;
        for(int i=0;i<k;i++)
        {
            int size=0;
            unordered_map<int,int> cnt;
            for(int j=i;j<n;j+=k)
            {
                ++cnt[nums[j]];
                ++size;
            }
            int min_value=*min_element(dp.begin(),dp.end());
            vector<int> dp2(maxn,min_value);
            for(int j=0;j<maxn;j++)
            {
                for(const auto &pr:cnt)
                    dp2[j]=min(dp2[j],dp[j^pr.first]-pr.second);
                dp2[j]+=size;
            }
            dp=move(dp2);
        }
        return dp[0];
    }
};