python - Another random tree generator

Question

I would like to construct a random tree: for each node, there a random number of leaves is created.

I am almost there, but I don't understand why recursion is done only at the first leaf of a branch.

This is my code:

# init, total_staff is the leaves counter
g = nx.DiGraph()
g.add_node(0)
total_staff = 0

# help function to create a group of leaves
# (the number of leaves will follow a given prob distribution)

def makeTeam():
    
    global total_staff
    
    # number of people reporting to root
    team_size = random.choices(team_population, weights = optimal_size_weight, k=1)[0]
    
    team = []
    for _ in range(team_size):
        total_staff += 1
        team_id = total_staff
        team.append(team_id)
    
    return team

# function to create a tree (representing an organisation)
def buildOrganisation(root):
    
    global total_staff
    if total_staff > 25:
        return g

    team = makeTeam()

    for team_id in team:
        print('creating sub ', root, team_id, team)
        g.add_edge(root, team_id)
        buildOrganisation(team_id) # this is the point I don't get: I expect recursion to be done for each team_id...
    
    return g

The result I have, though, is this: note that recursion is only at the first node of each branch, instead I want it for all nodes of the each branch.

Can you help me fix it and understand why ?

Answer 1

等待大神答复