【Leetcode】1611. Minimum One Bit Operations to Make Integers Zero（配数学证明）

题目地址：

https://leetcode.com/problems/minimum-one-bit-operations-to-make-integers-zero/

给定一个正整数$n$，将其视为二进制数。允许进行两种操作：
1、将其最低位取反；
2、如果左起第$i$位是$1$，并且第$i-1$位一直到第$0$位都是$0$，那就可以将第$i+1$位取反。
问要将其变为$0$至少需要多少步。

首先，对于任意$n$，它一定要先变成形如$1100...0$的数，才能继续变为$0$。这是因为要将$n$变为$0$，必须其中某一步是将最高位变为$0$，而最高位变为$0$必须要得到形如$1100...0$的数才能做。我们来证明两个命题：

1、对于$n=2^k$来说，至少需要$2^{k+1}-1$步。
数学归纳法：当$k=0$时显然。假设对于$k-1$的情况上述命题也正确，对于$k$的情况，由于$10...0$先要变成$110...0$，对于任意的能把$10...0$变成$110...0$的最短的操作序列，都不会改动最高位的$1$（如果改变了的话，说明中途已经到了$110...0$了，与最短矛盾），所以把$10...0$变成$110...0$的最短的操作序列其实就是把$0$变为$10...0$的操作序列（即只考虑除了最高位$1$的剩余数字）。而题目的两个操作实际上是在做异或运算，这个运算是可逆的，所以把$0$变为$10...0$的最短操作序列的逆序就是把$10...0$变为$0$的最短操作序列，从而根据归纳假设，其有$2^k-1$步，接着$110...0$变成$10...0$需要$1$步，然后再根据归纳假设，其经过$2^k-1$步变为$0$，所以总共步数就是$2^k-1+2^k=2^{k+1}-1$步。由数学归纳法，命题对于任意$k$都对。
2、将$n\to x$的最少步数等于$n\wedge x\to x\wedge x=0$的最少步数，这一点由异或运算满足结合律显然成立（左边同时做异或即可）。

我们回到原问题，设$f(n)$是答案。对于$n$而言，$n\to 110...0$需要的步数就等于$n\wedge 110...0\to 110...0\wedge 110...0=0$的步数，我们取$110...0$是能将$n$的最高位$1$通过异或消掉的数，令$k$是小于等于$n$的最大的$2$的幂（例如$n=10$的时候$k=2^3=8$，$n=17$的时候$k=2^4=16$），那么显然$k+k/2$就是形如$110...0$并且能将$n$的最高位$1$消掉的数。所以有$$f(n)=f(n\wedge 110...0)+1+k-1=f(n\wedge 110...0)+k$$只需要直接DFS就行了。因为每次参数的最高位都会被消掉，所以位数一定会越变越少，时间复杂度就是$O(\log n)$。代码如下：

public class Solution {
    public int minimumOneBitOperations(int n) {
        return dfs(n);
    }
    
    private int dfs(int n) {
        if (n == 0) {
            return 0;
        }
        
        int k = 1;
        while (k << 1 <= n) {
            k <<= 1;
        }
        
        return dfs(k ^ (k >> 1) ^ n) + k;
    }
}

时空复杂度$O(\log n)$。