C++徹底二叉樹的權值

 

 

#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>

int main(void)
{
    int n;

    printf("請輸入須要輸入的整數的數量：");
    scanf("%d", &n);

    // 肯定深度
    int count = 0, deep = 0;
    while(count < n)
    {
        deep++;
        count += pow(2.0, (deep - 1));
    }

    // 建立數組
    int **tree;
    tree = (int **)malloc((deep + 1) * sizeof(int *));
    for (int i = 0; i < deep + 1; i++)
    {
        tree[i] = (int *)malloc((deep + 1) * sizeof(int));
        memset(tree[i], 0, (deep + 1) * sizeof(int));
    }
    
    // 將輸入的值讓徹底二叉樹的規則輸入數組
    count = 0;
    for(int temp_deep = 1; temp_deep <= deep; temp_deep++)
    {
        for(int i = 1; i <= pow(2.0, temp_deep - 1); i++)
        {
            scanf("%d", &tree[temp_deep][i]);
            count++;
            if(count >= n) break;  // 當全部元素寫入以後推出輸入循環
        }
    }

    // 比較同一深度的節點權值之和
    int max_deep = 0, max_sum = 0;
    for(int temp_deep = 1; temp_deep <= deep; temp_deep++)
    {
        int sum = 0;
        for(int i = 1; i <= pow(2.0, temp_deep - 1); i++)
        {
            sum += tree[temp_deep][i];
        }
        if(sum > max_sum)
        {
            max_sum = sum;
            max_deep = temp_deep;
        }
    }

    printf("深度爲 %d 的節點權值之和最大\n", max_deep);

    printf("\n");
    system("pause");
}