如何创建树图和输入JSON来运行BFS (来自以下json)

{
  "name": "root",
  "children": [
    {
      "name": "child1",
      "children": [
        {"name": "grandchild1"},
        {"name": "grandchild2"}
      ]
    },
    {
      "name": "child2",
      "children": [
        {"name": "grandchild3"}
      ]
    }
  ]
}

import json
from collections import deque

# 定义树节点类
class TreeNode:
    def __init__(self, name):
        self.name = name
        self.children = []

# 从JSON创建树图
def create_tree_from_json(json_data):
    root = TreeNode(json_data['name'])
    queue = deque([root])
    
    while queue:
        current_node = queue.popleft()
        for child_data in json_data.get('children', []):
            child_node = TreeNode(child_data['name'])
            current_node.children.append(child_node)
            queue.append(child_node)
            json_data = child_data  # 更新json_data为子节点的数据
    
    return root

# BFS遍历函数
def bfs_traversal(root):
    if root is None:
        return []
    
    queue = deque([root])
    traversal_result = []
    
    while queue:
        current_node = queue.popleft()
        traversal_result.append(current_node.name)
        for child in current_node.children:
            queue.append(child)
    
    return traversal_result

# JSON数据
json_data = {
  "name": "root",
  "children": [
    {
      "name": "child1",
      "children": [
        {"name": "grandchild1"},
        {"name": "grandchild2"}
      ]
    },
    {
      "name": "child2",
      "children": [
        {"name": "grandchild3"}
      ]
    }
  ]
}

# 创建树图并进行BFS遍历
root_node = create_tree_from_json(json_data)
result = bfs_traversal(root_node)
print(result)  # 输出应该是 ['root', 'child1', 'child2', 'grandchild1', 'grandchild2', 'grandchild3']