AcWing4313. 满二叉树等长路径(递归+贪心）

题目链接

思路

若要使所有叶子节点到根节点距离相等,那么首先应使左子树的所有叶子节点到左子树根节点距离相等(设为l),使右子树所有叶子节点到右子树根节点距离相等(设为r),然后再将左右子树的所有叶子节点到根节点的距离变为$max(l,r)$,所以应增加的距离应为$|l-r|$,可利用递归对每一层实现这个过程
代码

#include<iostream>
#include<algorithm>
#include<stack>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
#include<cmath>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
typedef pair<int,int> P;
const int N=1e4+10;
const double esp=1e-5;
const ll mod=1e9+7;

int n,a[N],ans;

int dfs(int x){
    if(x>=(1<<n))
        return 0;
    int l=dfs(x*2)+a[x*2];
    int r=dfs(x*2+1)+a[x*2+1];
    ans+=abs(l-r);
    return max(l,r);
}
int main()
{
    cin>>n;
    for(int i=2;i<=(1<<n+1)-1;i++)
        cin>>a[i];
    dfs(1);
    cout<<ans<<endl;
    return 0;
}