검 지 offer - 이 진 트 리 의 다음 노드

제목 설명:
이 진 트 리 와 그 중 하 나 를 지정 합 니 다. 중간 순 서 를 옮 겨 다 니 는 다음 결점 을 찾 아 되 돌려 주 십시오.주의 하 세 요. 나무의 결점 은 좌우 자 결점 뿐만 아니 라 아버지 결점 을 가리 키 는 지침 도 포함 되 어 있 습 니 다.
생각:
중 서 를 옮 겨 다 니 는 특징 에 따라 한 노드 의 다음 노드 를 찾 으 려 면 세 가지 상황 이 있 습 니 다. 1. 오른쪽 나무 가 있 습 니 다. 이 때 는 오른쪽 아 이 를 다음 에 옮 겨 다 니 는 (찾 으 려 는 것 이 아니 라) 노드 로 한 다음 에 이 노드 의 왼쪽 나무 (있 으 면) 를 따라 출발 하여 잎 노드 를 만 날 때 까지 이 잎 노드 는 다음 에 찾 아야 할 노드 입 니 다.2. 오른쪽 나무 가 없 으 면 이 노드 가 아버지 노드 의 왼쪽 아이 인지 판단 하고 다음 에 찾 아야 할 노드 는 아버지 노드 이다.3. 만약 에 아버지 노드 의 왼쪽 아이 가 아니라면 아버지 노드 를 다음 에 옮 겨 다 니 는 노드 로 하고 위로 거 슬러 올 라 가 아버지 노드 에 아버지 노드 가 없고 아버지 노드 가 아버지 노드 의 아버지 노드 의 왼쪽 아 이 를 찾 을 때 까지 한다.이 세 가지 상황 을 종합 하면 이 진 트 리 의 임의의 노드 의 다음 노드 를 찾 을 수 있다.
코드 구현:
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}

public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        TreeLinkNode curNode = null;
        //   :        
        if(pNode.right != null){
            curNode = pNode.right;
            while(curNode.left != null) curNode = curNode.left;
            return curNode;
        }
        //   :             
        if(pNode.next == null) return null;
        if(pNode == pNode.next.left){
            return pNode.next;
        }
        //   :       ,                  
        curNode = pNode.next;
        while(curNode.next != null){
            if(curNode == curNode.next.left){
                return curNode.next;
            }
            //        
            curNode = curNode.next;
        }
        return null;
    }
}

좋은 웹페이지 즐겨찾기