题目:343. 整数拆分、96.不同的二叉搜索树
参考链接:代码随想录
343. 整数拆分
思路:五部曲来走。dp数组,dp[i]用于记录拆i得到的最大乘积和,我们要求的也就是dp[n];递推公式,我们想拆分i,可以拆成i-1和1,这样dp[i]就是dp[i-1]或者就是i-1,也可以拆成i-2和2,这样就是2dp[i-2]和2(i-2)取最大值,综合讨论一下就是拆成i-j和j,然后比较dp[i-j]*j和(i-j)*j的最大值,这里因为dp[i-j]必须要进行至少拆成两个数,所以没有包含i-j,故我们还需要比较i-j,递推公式就是dp[i]=max(dp[i-j]*j,(i-j)*j)
;初始化,这里0和1根据题目无法拆分,也就无需初始化这两个,直接初始化dp[2]=1;遍历顺序和举例略。时间复杂度O(n^2)。
class Solution {
public:
int integerBreak(int n) {
vector<int> dp(n+1);//其实这里也不用初始化为0
dp[2]=1;
for(int i=3;i<=n;i++){
for(int j=1;j<=i-2;j++){//最多只能拆i-2个,这样剩下2,如果再拆得只剩1,dp[1]就没有意义了
dp[i]=max(dp[i],max(dp[i-j]*j,(i-j)*j));//注意C++的max只能有两个参数
}//更新dp[i]需要一个循环遍历
}
return dp[n];
}
};
看标答发现可以优化,即拆分的时候拆到一半就可以了,根据数学推导,拆分成相同的子数乘积比较大,故拆得超过一半必定不可能是最大值,当然我们写代码的时候上面其实就可以了:
class Solution {
public:
int integerBreak(int n) {
vector<int> dp(n + 1);
dp[2] = 1;
for (int i = 3; i <= n ; i++) {
for (int j = 1; j <= i / 2; j++) {
dp[i] = max(dp[i], max((i - j) * j, dp[i - j] * j));
}
}
return dp[n];
}
};
贪心方法需要数学证明略。
96.不同的二叉搜索树
思路:本题一看根本没有思路,主要要了解BST,就是左子树都小于头,右子树都大于头,而且每一个子树都是BST。使用dp五部曲。首先是dp[i]表示i个节点组成的不同BST个数;然后是递推公式,我们可以举例考虑3的情况和2来组合,首先是1为头结点,3可以插到1和2中间,也可以插到2右边,两种,然后是2为头节点,这时只有一种,最后是3为头结点,也是两种,共有5种。仔细对比可以发现规律,1为头时,两个节点的布局和n为2的时候一样,因为左右都是两个节点,3为头也是如此,2为头的时候,左右各一个节点,布局同n=1的情况。dp[3]就是将1,2,3分别为头结点的BST数量相加。1为头,左子树节点个数为0,右子数节点个数为2,即dp[0]*dp[2],2为头,dp[1]*dp[1],3为头dp[2]*dp[0]。
故对dp[i],我们也考虑1到i为头节点的数量相加,dp[i]=dp[0]*dp[i-1]+dp[1]*dp[i-2]+…+dp[i-1]*dp[i],这是重新回顾dp[i]的定义,可以定义为i个不同元素组成BST的数量;初始化,dp[0]=1,dp[1]=1,其他都可以推出来;遍历顺序从前往后;举例略。时间复杂度O(n^2)。
class Solution {
public:
int numTrees(int n) {
vector<int> dp(n+1,0);
dp[0]=1;
dp[1]=1;
for(int i=2;i<=n;i++){
for(int j=0;j<=i-1;j++){
dp[i]+=dp[j]*dp[i-1-j];
}
}
return dp[n];
}
};