CF Om Nom and Dark Park

分析:题意是一颗满二叉树,输入树的高度n,按树的顺序输入两个结点之间的权,问你最少添加多少权,可以使根结点到每个叶子结点之间的权值的和相等。因为是满二叉树,所以可以用数组来储存,结点i的左孩子为2*i,右孩子为2*i+1.。所以从最后一个有孩子的结点往前循环到根结点,计算每棵子树的左孩子和有孩子的差,差值加到权值较小的树上。

# include <stdio.h>
# include <math.h>
  int s[5000][2];//s[i][0]代表和左孩子之间的权值,s[i][1]代表右孩子
  int main()
  {
	  int i,j,n,v,t=0,k,a,b;
	  scanf("%d",&n);
	  v=pow(2,n+1);
	  v--;
	  for(i=1;i<=v;i++)
		  s[i][0]=s[i][1]=0;
	  for(i=1;i<=v/2;i++)
		  scanf("%d%d",&s[i][0],&s[i][1]);
	  for(i=v/2,t=0;i>=1;i--)
	  {
		  for(j=i,a=0;j<=v;j=j*2)
			  a+=s[j][0];//a计算根结点到左边叶子结点的权值
		  for(j=i,b=0;j<=v;j=j*2+1)
			  b+=s[j][1];//b计算根结点到右边叶子结点的权值
		  k=a-b;
		  if(k<0)
		  {
			  k=-k;
			  s[i][0]+=k;
		  }
		  else
			  s[i][1]+=k;
		  t=t+k;
	  }
	  printf("%d\n",t);
	  return 0;
  }

你可能感兴趣的:(ACM,CF)