【Leetcode】Populating Next Right Pointers in Each Node II

2053 단어
제목 링크:https://leetcode.com/problems/populating-next-right-pointers-in-each-node-ii/
제목:
Follow up for problem "Populating Next Right Pointers in Each Node".
What if the given tree could be any binary tree? Would your previous solution still work?
Note:
You may only use constant extra space.
For example, Given the following binary tree,
         1
       /  \
      2    3
     / \    \
    4   5    7

After calling your function, the tree should look like:
         1 -> NULL
       /  \
      2 -> 3 -> NULL
     / \    \
    4-> 5 -> 7 -> NULL

생각:
1、"Populating Next Right Pointers in Each Node"를 사용합니다.의 알고리즘은 여전히 유효하지만 공간 복잡도는 O(n)
알고리즘 1:
	public void connect(TreeLinkNode root) {
		List<List<TreeLinkNode>> lists = levelOrder(root);
		for (List<TreeLinkNode> list : lists) {
			for (int i = 0; i < list.size() - 1; i++) {
				list.get(i).next = list.get(i + 1);
			}
			list.get(list.size() - 1).next = null;
		}
	}

	/**
	 *  
	 */
	public List<List<TreeLinkNode>> levelOrder(TreeLinkNode root) {
		int height = heightTree(root);
		List<List<TreeLinkNode>> lists = new ArrayList<List<TreeLinkNode>>();
		for (int i = 1; i <= height; i++) {
			List<TreeLinkNode> list = new ArrayList<TreeLinkNode>();
			list = kLevelNumber(root, 1, list, i);
			lists.add(list);
		}
		return lists;
	}

	/***
	 * kk ,height  
	 */
	public List<TreeLinkNode> kLevelNumber(TreeLinkNode p, int height, List<TreeLinkNode> list, int kk) {
		if (p != null) {
			if (height == kk) {
				list.add(p);
			}
			list = kLevelNumber(p.left, height + 1, list, kk);
			list = kLevelNumber(p.right, height + 1, list, kk);
		}
		return list;
	}

	public int heightTree(TreeLinkNode p) {
		if (p == null)
			return 0;
		int h1 = heightTree(p.left);
		int h2 = heightTree(p.right);
		return h1 > h2 ? h1 + 1 : h2 + 1;
	}

좋은 웹페이지 즐겨찾기