题目链接:
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