题目
假设你有一个特殊的键盘包含下面的按键:
Key 1: (A):在屏幕上打印一个 ‘A’。
Key 2: (Ctrl-A):选中整个屏幕。
Key 3: (Ctrl-C):复制选中区域到缓冲区。
Key 4: (Ctrl-V):将缓冲区内容输出到上次输入的结束位置,并显示在屏幕上。
现在,你只可以按键 N 次(使用上述四种按键),请问屏幕上最多可以显示几个 'A’呢?
样例 1:
输入: N = 3
输出: 3
解释:
我们最多可以在屏幕上显示三个’A’通过如下顺序按键:
A, A, A
样例 2:
输入: N = 7
输出: 9
解释:
我们最多可以在屏幕上显示九个’A’通过如下顺序按键:
A, A, A, Ctrl A, Ctrl C, Ctrl V, Ctrl V
注释:
1 <= N <= 50
结果不会超过 32 位有符号整数范围。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4-keys-keyboard
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
状态转移方程
如果要达到最大肯定是先输入一定次数的A,然后三连(Ctrl-A, Ctrl-C, Ctrl-V * n)。
因为,只输入A一次只能增加一,而Ctrl-V可以增加前面复制的A的个数,肯定大于1。所以有这样的事实
输入序列为
A * n, Ctrl-A, Ctrl-C, Ctrl-V * n
所以dp[i],
有两个来源
- dp[i-1] + 1(直接输入A)
- dp[j] * (i-j-2+1) 从[0, j]一直输入A,后面复制,复制的次数为(i - j - 2),这两次为 Ctrl-A, Ctrl-C,最后还需要加上复制之前原始输入了的A的数量,所以要 + 1
// 直接输入A
dp[i]=dp[i-1] + 1;
for (int j = 0; j < i - 2; j++) {//j < i - 2是因为有两次Ctrl-A, Ctrl-C,不可避免
// 复制而来
dp[i] = Math.max(dp[i], dp[j] * (i - j - 2 + 1));
}
Base
由状态转移方程可知,只依赖前一个
dp[0]=1
答案
class Solution {
public int maxA(int N) {
int[] dp = new int[N + 1];
dp[0] = 0;
dp[1] = 1;
for (int i = 2; i < N + 1; i++) {
dp[i] = dp[i - 1] + 1;
for (int j = 2; j < i; j++) {
dp[i] = Math.max(dp[i], dp[j - 2] * (i - j + 1));
}
}
return dp[N];
}
}