1. 程式人生 > >牛客國慶集訓派對Day2 F-平衡二叉樹(簡單dp)

牛客國慶集訓派對Day2 F-平衡二叉樹(簡單dp)

思路來源

錢神

錢神如是說(樣例n=4 d=1):

對於高度為5,滿足d=1的樹來說,只需要左子樹高度4,右子樹高度3 就可以滿足d=1 右子樹只要滿足右左子樹高度3且d=1,右右子樹高度2且d=1 然後遞迴的去求 最後發現其實可以遞推。

題解

轉移方程

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;
}