牛客国庆集训派对Day2 F-平衡二叉树(简单dp)

本文探讨了一种使用动态规划解决深度相差不超过d的完全二叉树最小节点数问题的方法。通过分析不同深度下树的结构特性,给出了递推公式,并提供了两种代码实现方案,一种使用DFS记忆化搜索(WA),另一种使用循环(AC)。文章对比了两种方法的优劣,强调了正确选择算法的重要性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

思路来源

钱神

钱神如是说(样例n=4 d=1):

对于高度为5,满足d=1的树来说,只需要左子树高度4,右子树高度3
就可以满足d=1
右子树只要满足右左子树高度3且d=1,右右子树高度2且d=1
然后递归的去求
最后发现其实可以递推。

题解(转移方程)

其中i是树的节点数,a[i][d]对应的是树的总结点个数,

这是在深度相差不超过d之下的最小节点个数。

其实预处理一下所有的d也可以,毕竟60*60。

a[i][d]=1+a[i-1][d]+a[i-1-d][d];(i>=1+d)

a[i][d]=i;(i<=d)

一个简单dp,然而自己太菜了。

心得

很迷的一道题,

dfs记忆化搜索会WA样例。

循环就过了,gg。

代码①(WA代码)

#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#include <bitset>
const int INF=0x3f3f3f3f;
const int maxn=1e5+5;
const int mod=1e9+7;
const double eps=1e-7;
typedef long long ll;
#define vi vector<int>
#define si set<int>
#define pii pair<ll,ll>
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
ll n,d,a[100][100];
ll fib(ll depth,ll d)
{
    int ans;
    if(depth<=0)return 0;
    if(a[depth][d]!=-1)return a[depth][d];
    if(d==0)ans=0;
    else if(depth<=d)ans=depth;
    else ans=1+fib(depth-1,d)+fib(depth-1-d,d);
    a[depth][d]=ans;
    //printf("depth:%d,d:%d,ans:%d\n",depth,d,ans);
    return ans;
}
int main()
{
    scll(n),scll(d);
    mem(a,-1);
    if(n<=1||d==0)puts("0");
    else printf("%lld\n",((ll)1<<(n-1))-1-fib(n-1-d,d));
    return 0;
}

代码②(AC代码)

#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#include <bitset>
const int INF=0x3f3f3f3f;
const int maxn=1e5+5;
const int mod=1e9+7;
const double eps=1e-7;
typedef long long ll;
#define vi vector<int>
#define si set<int>
#define pii pair<ll,ll>
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
ll n,d,a[100][100];
int main()
{
    scanf("%lld%lld",&n,&d);
    for(int i=1;i<=n-1-d;++i)
    {
        if(i<=d)a[i][d]=i;
        else
        {
         ll tmp=i-1-d<=0?0:a[i-1-d][d];
         a[i][d]=1+a[i-1][d]+tmp;
        }
        //printf("%lld:%lld\n",i,a[i][d]);
    }
    if(n<=1||d==0)puts("0");
    else
    {
     ll tmp=n-1-d<=0?0:a[n-1-d][d];
     printf("%lld\n",((ll)1<<(n-1))-1-tmp);
    }
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小衣同学

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值