Codeforces ZeptoLab Code Rush 2015 B. Om Nom and Dark Park DFS

题目链接：

http://codeforces.com/contest/526/problem/B

题意：

n表示几层边
使从根走到每个叶子所要的花费都相同，问最少得增加多少边权？

题解：

把每个点与父节点的边 和 这个点 编号相同， 这样就使得每条边的编号是它儿子边的编号的2倍
DFS，自底往上，预处理前缀和
然后ans加上左右叶子的差值，然后一直跑DFS就是了

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define mp(x,y) make_pair(x,y)
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e4+10;
ll a[maxn];
ll n,k,dp[maxn];

void read_dfs(int x){
    if(x > k) return ;

    read_dfs(2*x);
    read_dfs(2*x+1);
    a[2*x] += max(a[2*x * 2], a[2*x*2 + 1]); // 对边累加，前缀和
    a[2*x+1] += max(a[(2*x+1)*2],a[(2*x+1)*2+1]);
}

ll dfs(ll x){
    if(dp[x]) return dp[x];
    if(x > k) return 0;

    dp[x] = dfs(x*2)+dfs(x*2+1) + abs(a[x*2]-a[x*2+1]); // 对x这个点更新答案
    return dp[x];
}

int main(){
    cin >> n;int c=0;
    k = pow(2,n);
    for(int i=1; i<=n; i++)
        for(int j=0; j<pow(2,i); j++){
            int p = pow(2,i)+j;
            cin >> a[p];
        }

    read_dfs(1);

    cout << dfs(1) << endl;

    return 0;
}

//http://codeforces.com/contest/526/problem/B