节点

Question

我正在HackerRank进行编程练习。 我制作了这样的数据结构代码。但我无法弄清楚一个计划如何实现程序来总结树节点的价值。 你能给我一些关于我代码的建议吗？ 如果我在数据结构中出现了一些错误，请告诉我这个问题。

谢谢。

[HackerRank练习]

https://www.hackerrank.com/challenges/cut-the-tree

[代码]

我写了这个程序，但我怎么能不能总结树的价值。

class Tree:
    """
    Original Tree Structure which is representing the Input
    """
    def __init__(self, value, position):
        self.value = value
        self.position = position
        self.connection = []

    def set_connection(self, connection):
        self.connection.append(connection)

    def set_value(self, value):
        self.connection = value

    def get_value(self, position):
        return self.value

    def get_sum(self):
        return self.value

    def get_position(self):
        return self

if __name__ == "__main__":

    # ----- Initializing ------
    N = int(input())
    values = input().split()
    T = []
    a = 0
    for value in values:
        T.append(Tree(int(value), a))
        a += 1

    input_list = []
    for i in range(N-1):
        in_con = input().split()
        T[int(in_con[0])-1].set_connection(int(in_con[1])-1)
        T[int(in_con[1])-1].set_connection(int(in_con[0])-1)
        input_list.append(in_con)

    # ----- Start solve problem -----
    ~~~~~~~~~~snip~~~~~~~~~~~~~~~~

Answer 1

我不打算编写完整的代码，但你可以遵循这个伪代码：

java.io.Reader.*;