二分木の高さを求めたいのですが、再帰的なロジックを間違えてしまっています。

Question

###前提・実現したいこと
再帰的なsetHeight()で高さを求めたいのですがロジックを間違えてしまってしまいました。
どのように修正すればいいのでしょうか？

二分木で
９つの要素０から８までで
left左要素、right右要素
parent親要素、depth深さ、height高さ

０は１と４を持ち ：０は高さ３
１は２と３を持ち、４は５と８を持ち、 ：４は高さ２
５は６と７を持つ ：６と７は高さ０、５は高さ１

ときのrootの0の高さを求めるなど、各要素の高さを求めたいのですが、
高さを求める設定するsetHeight()のロジックがおかしいようです。
どのように修正いすればいいでしょうか？

###発生している問題・エラーメッセージ

エラーメッセージ

###該当のソースコード

import java.util.*;

class Node{
	int id;
	int parent, left, right, sibling, depth, height;

	public Node(int id){
		this.id = id;
		this.parent = -1;
		this.left = -1;
		this.right = -1;
		this.sibling = -1;
		this.depth = -1;
		this.height = -1;
	}
}

class Nodes{
	Node[] nodes;

	public void setDepth(int id, int d){
		int l = nodes[id].left;
		int r = nodes[id].right;
		if(l != -1)
			setDepth(l, d+1);
		if(r != -1)
			setDepth(r, d+1);
		nodes[id].depth = d;
	}

	public int setHeight(int id){
		if(id == -1)
			return 0;
		int l = nodes[id].left;
		int r = nodes[id].right;
		nodes[id].height = Math.max(setHeight(l), setHeight(r));
		return nodes[id].height++;
	}

	public void printNode(int id){
		StringBuffer sb = new StringBuffer();
		int degree = 0;
		if(nodes[id].left != -1) degree++;
		if(nodes[id].right != -1) degree++;
		sb.append("node " + id + ": parent = " + nodes[id].parent 
					+ ", sibling = " + nodes[id].sibling + ", degree = " + degree 
					+ ", depth = " + nodes[id].depth 
					+ ", height = " + nodes[id].height + ", ");
		if(nodes[id].parent == -1) sb.append("root\n");
		else if(degree == 0) sb.append("leaf\n");
		else sb.append("internal node\n");
		System.out.print(sb);
	}
}

public class Main{

	public static void main(String[] args){
		Scanner scan = new Scanner(System.in);
		int n = scan.nextInt();

		Nodes mynodes = new Nodes();
		Node[] nodes = new Node[n];
		mynodes.nodes = nodes;
		for(int i = 0; i < n; i++)
			nodes[i] = new Node(i);
		for(int i = 0; i < n; i++){
			int id = scan.nextInt();
			int left = scan.nextInt();
			int right = scan.nextInt();
			if(left != -1) nodes[left].parent = id;
			if(right != -1) nodes[right].parent = id;
			nodes[id].left = left;
			nodes[id].right = right;
			if(left != -1 && right != -1){
				nodes[left].sibling = right;
				nodes[right].sibling = left;
			}
		}
		int root = 0;
		while(nodes[root].parent != -1)
			root = nodes[root].parent;
		mynodes.setDepth(root, 0);
		nodes[root].height = mynodes.setHeight(root);
		
		for(int i = 0; i < n; i++)
			mynodes.printNode(i);
	}
}

###試したこと
課題に対してアプローチしたことを記載してください

###補足情報(言語/FW/ツール等のバージョンなど)

public int setHeight(int id){
	if(id == -1)
		return -1;
	int l = nodes[id].left;
	int r = nodes[id].right;
	nodes[id].height = Math.max(setHeight(l), setHeight(r)) + 1;
	return nodes[id].height;
}

これでいいのかな。

Answer 1

public int setHeight(int id){
if(id == -1)
return -1;
int l = nodes[id].left;
int r = nodes[id].right;
nodes[id].height = Math.max(setHeight(l), setHeight(r)) + 1;
return nodes[id].height;
}

これでいいのかな。