📌  相关文章
📜  在 n 叉树中具有最大直接子节点和自身的节点

📅  最后修改于: 2022-05-13 01:57:18.112000             🧑  作者: Mango

在 n 叉树中具有最大直接子节点和自身的节点

给定一个N-Ary 树,找到并返回所有子节点和节点本身的数据总和最大的节点。总之,要取节点本身的数据和它的直接子节点的数据。
例如在给定的树中,

maxSum 节点 = 4,最大和为 28

这个想法是我们将维护一个包含最大和的整数变量maxsum ,以及一个指向具有最大和的节点的resnode节点指针。
遍历树并在currsum中维护其所有直接子节点的根和数据的总和
整数变量并相应地更新maxsum变量。

C++
// CPP program to find the node whose children
// and node sum is maximum.
#include 
using namespace std;
 
// Structure of a node of an n-ary tree
struct Node {
    int key;
    vector child;
};
 
// Utility function to create a new tree node
Node* newNode(int key)
{
    Node* temp = new Node;
    temp->key = key;
    return temp;
}
 
// Helper function to find the node
void maxSumUtil(Node* root, Node** resNode,
                int* maxsum)
{
    // Base Case
    if (root == NULL)
        return;
 
    // curr contains the sum of the root and
    // its children
    int currsum = root->key;
 
    // total no of children
    int count = root->child.size();
 
    // for every child call recursively
    for (int i = 0; i < count; i++) {
        currsum += root->child[i]->key;
        maxSumUtil(root->child[i], resNode, maxsum);
    }
 
    // if curr is greater than sum, update it
    if (currsum > *maxsum) {
 
        // resultant node
        *resNode = root;
        *maxsum = currsum;
    }
    return;
}
 
// Function to find the node having max sum of
// children and node
int maxSum(Node* root)
{
    // resultant node with max sum of children
    // and node
    Node* resNode;
 
    // sum of node and its children
    int maxsum = 0;
 
    maxSumUtil(root, &resNode, &maxsum);
 
    // return the key of resultant node
    return resNode->key;
}
 
// Driver program
int main()
{
    /*   Let us create below tree
    *              1
    *          /   |  \
    *         2   3   4
    *        / \    / |  \ \
    *       5   6  7  8  9  10
    */
 
    Node* root = newNode(1);
    (root->child).push_back(newNode(2));
    (root->child).push_back(newNode(3));
    (root->child).push_back(newNode(4));
    (root->child[0]->child).push_back(newNode(5));
    (root->child[0]->child).push_back(newNode(6));
    (root->child[2]->child).push_back(newNode(5));
    (root->child[2]->child).push_back(newNode(6));
    (root->child[2]->child).push_back(newNode(6));
 
    cout << maxSum(root) << endl;
 
    return 0;
}


Java
// Java program to find the node whose children
// and node sum is maximum.
import java.util.*;
 
class GFG
{
     
// Structure of a node of an n-ary tree
static class Node
{
    int key;
    Vector child;
    Node()
    {
        child = new Vector();
    }
};
 
// Utility function to create a new tree node
static Node newNode(int key)
{
    Node temp = new Node();
    temp.key = key;
    return temp;
}
 
static int maxsum;
 
// resultant node with max sum of children
// and node
static Node resNode;
 
// Helper function to find the node
static void maxSumUtil(Node root)
{
    // Base Case
    if (root == null)
        return;
 
    // curr contains the sum of the root and
    // its children
    int currsum = root.key;
 
    // total no of children
    int count = root.child.size();
 
    // for every child call recursively
    for (int i = 0; i < count; i++)
    {
        currsum += root.child.get(i).key;
        maxSumUtil(root.child.get(i));
    }
 
    // if curr is greater than sum, update it
    if (currsum > maxsum)
    {
 
        // resultant node
        resNode = root;
        maxsum = currsum;
    }
    return;
}
 
// Function to find the node having max sum of
// children and node
static int maxSum(Node root)
{
     
    // sum of node and its children
    int maxsum = 0;
 
    maxSumUtil(root);
 
    // return the key of resultant node
    return resNode.key;
}
 
// Driver code
public static void main(String args[])
{
    /* Let us create below tree
                1
            / | \
            2 3 4
        / \ / | \ \
        5 6 7 8 9 10
    */
 
    Node root = newNode(1);
    (root.child).add(newNode(2));
    (root.child).add(newNode(3));
    (root.child).add(newNode(4));
    (root.child.get(0).child).add(newNode(5));
    (root.child.get(0).child).add(newNode(6));
    (root.child.get(2).child).add(newNode(5));
    (root.child.get(2).child).add(newNode(6));
    (root.child.get(2).child).add(newNode(6));
 
    System.out.print( maxSum(root) );
}
}
 
// This code is contributed by Arnab Kundu


Python3
# Python3 program to find the node
# whose children and node sum is maximum.
 
# Structure of a node of an n-ary tree
class Node:
     
    def __init__(self, key):
        self.key = key
        self.child = []
 
# Helper function to find the node
def maxSumUtil(root, resNode, maxsum):
 
    # Base Case
    if root == None:
        return
 
    # curr contains the sum of the root
    # and its children
    currsum = root.key
 
    # total no of children
    count = len(root.child)
 
    # for every child call recursively
    for i in range(0, count):
        currsum += root.child[i].key
        resNode, maxsum = maxSumUtil(root.child[i],
                                     resNode, maxsum)
 
    # if curr is greater than sum,
    # update it
    if currsum > maxsum:
 
        # resultant node
        resNode = root
        maxsum = currsum
     
    return resNode, maxsum
 
# Function to find the node having
# max sum of children and node
def maxSum(root):
 
    # resultant node with max
    # sum of children and node
    resNode, maxsum = Node(None), 0
    resNode, maxsum = maxSumUtil(root, resNode,
                                       maxsum)
 
    # return the key of resultant node
    return resNode.key
 
# Driver Code
if __name__ == "__main__":
 
    root = Node(1)
    (root.child).append(Node(2))
    (root.child).append(Node(3))
    (root.child).append(Node(4))
    (root.child[0].child).append(Node(5))
    (root.child[0].child).append(Node(6))
    (root.child[2].child).append(Node(5))
    (root.child[2].child).append(Node(6))
    (root.child[2].child).append(Node(6))
 
    print(maxSum(root))
 
# This code is contributed by Rituraj Jain


C#
// C# program to find the node whose children
// and node sum is maximum
using System;
using System.Collections.Generic;
 
class GFG
{
     
// Structure of a node of an n-ary tree
public class Node
{
    public int key;
    public List child;
    public Node()
    {
        child = new List();
    }
};
 
// Utility function to create a new tree node
static Node newNode(int key)
{
    Node temp = new Node();
    temp.key = key;
    return temp;
}
 
static int maxsum;
 
// resultant node with max sum of children
// and node
static Node resNode;
 
// Helper function to find the node
static void maxSumUtil(Node root)
{
    // Base Case
    if (root == null)
        return;
 
    // curr contains the sum of the root and
    // its children
    int currsum = root.key;
 
    // total no of children
    int count = root.child.Count;
 
    // for every child call recursively
    for (int i = 0; i < count; i++)
    {
        currsum += root.child[i].key;
        maxSumUtil(root.child[i]);
    }
 
    // if curr is greater than sum, update it
    if (currsum > maxsum)
    {
 
        // resultant node
        resNode = root;
        maxsum = currsum;
    }
    return;
}
 
// Function to find the node having max sum of
// children and node
static int maxSum(Node root)
{
     
    // sum of node and its children
    int maxsum = 0;
 
    maxSumUtil(root);
 
    // return the key of resultant node
    return resNode.key;
}
 
// Driver code
public static void Main(String []args)
{
    /* Let us create below tree
                1
            / | \
            2 3 4
        / \ / | \ \
        5 6 7 8 9 10
    */
 
    Node root = newNode(1);
    (root.child).Add(newNode(2));
    (root.child).Add(newNode(3));
    (root.child).Add(newNode(4));
    (root.child[0].child).Add(newNode(5));
    (root.child[0].child).Add(newNode(6));
    (root.child[2].child).Add(newNode(5));
    (root.child[2].child).Add(newNode(6));
    (root.child[2].child).Add(newNode(6));
 
    Console.Write( maxSum(root) );
}
}
 
// This code has been contributed by 29AjayKumar


Javascript


输出:

4