【Lintcode】867. 4 Keys Keyboard（配数学证明）

题目地址：

https://www.lintcode.com/problem/4-keys-keyboard/description

给定一个屏幕，每次可以执行下面的操作：
1、在屏幕上添加一个字母A；
2、选中笔记本所有A（记为CtrlA）；
3、将选中的A存入缓冲区（记为CtrlC）；
4、将缓冲区里的字符全部追加到屏幕上（记为CtrlV）。
问执行$n$次操作后屏幕上最多能有多少个A。

思路是动态规划。设$f[i]$代表经过$i$次操作能得到的最多A的个数（规定$f[0]=0$，意思是什么也不做）。我们先证明能得到最多$A$的操作步骤一定可以写成以CtrlA，CtrlC后面接若干个CtrlV结尾，或者以步骤$1$结尾。
当$n=1,2,3,4,5,6$的时候结论成立。如果最优解最后一步是步骤$1$，那么结论已经成立。如果最优解最后一步是CtrlV（那么这次粘贴的字符数一定大于等于$1$，否则能用步骤$1$替代），则向前找到第一个不等于CtrlV的操作（这个操作一定存在，下同），如果该操作是步骤$1$，则可以将其替换为CtrlV得到最优解，继续向前找；显然紧接着CtrlV前面的不能是CtrlA；如果紧接着的是CtrlC，则继续向前找，显然CtrlC只能连续出现一次（否则可以将靠后的多余的CtrlC替换为CtrlV得到更优解，矛盾），并且CtrlC紧接着前面不能是CtrlV（否则就可以将这个CtrlC替换为CtrlV得到更优解，矛盾），如果遇到了步骤$1$，那也可以将这个CtrlC替换为CtrlV得到更优解，矛盾，所以紧接着CtrlC前面的必然是CtrlA，从而结论正确。上面其实就是证明了，在连续多个CtrlV之前的那个步骤一定是CtrlC，然后紧接着CtrlC前面的一定是CtrlA，否则就能得到更优解。

证明完上面的结论之后，就可以知道最优解一定是$f[i]=f[j]\times (i-j-1)$的形式，其中$f[j]$表示当最后一次执行CtrlA的时候，屏幕上有多少个字符，那么这次CtrlA实际上是第$j+1$次操作，从而之后执行CtrlV的操作次数是$i-j-2$，所以得到的字符个数就是$f[i]=f[j]\times (i-j-1)$。只需要枚举$j$即可。代码如下：

public class Solution {
    /**
     * @param N: an integer
     * @return: return an integer
     */
    public int maxA(int N) {
        // write your code here
        int[] dp = new int[N + 1];
        dp[0] = 0;
    
        for (int i = 1; i <= N; i++) {
        	// 枚举以步骤1结尾的情况
            dp[i] = dp[i - 1] + 1;
            // 枚举以CtrlV结尾的情况，j枚举的是CtrlA的操作次数的编号减去1
            for (int j = 0; j + 1 < i; j++) {
                dp[i] = Math.max(dp[i], dp[j] * (i - j - 1));
            }
        }
        
        return dp[N];
    }
}

时间复杂度$O(n^2)$。