776. Split BST

[dxxsxsxkx]

問題

https://www.lintcode.com/problem/847/description

LintCode の問題なので、LeetCode の同名の問題と若干要件が違います。

次の問題

3. Longest Substring Without Repeating Characters

[mamo3gr]

left_count == right_count の場合での処理の行数が多いので、先に読ませると消化しやすいように思いました。

Suggested change

	if (left_count > right_count) {
	return split_trees.first;
	} else if (left_count < right_count) {
	return split_trees.second;
	} else {
	if (!split_trees.first) {
	return split_trees.second;
	}
	if (!split_trees.second) {
	return split_trees.first;
	}
	return (split_trees.first->val > split_trees.second->val) ?
	split_trees.first :
	split_trees.second;
	}
	if (left_count == right_count) {
	if (!split_trees.first) {
	return split_trees.second;
	}
	if (!split_trees.second) {
	return split_trees.first;
	}
	return (split_trees.first->val > split_trees.second->val) ?
	split_trees.first :
	split_trees.second;
	}
	if (left_count > right_count) {
	return split_trees.first;
	} else {
	return split_trees.second;
	}

出題サイトをざっと見た感じ制約を見つけられなかったのですが、left_count == right_count かつ split_trees.first も split_trees.second も nullptr の場合は無いんでしたっけ（おそらく root が nullptr の場合のみ？）。

[mamo3gr]

left_count == right_count かつ split_trees.first も split_trees.second も nullptr の場合は、split_trees.second = nullptr が返るので問題ないですね。

[mamo3gr]

first -> left, second -> right の変換を脳内で都度やる必要があり、先に置き換えてもらえると読み手としては助かります。

Suggested change

	auto split = splitHelper(root, v);
	int left_count = countNode(split.first);
	int right_count = countNode(split.second);
	auto split = splitHelper(root, v);
	TreeNode* left = split.first;
	TreeNode* right = split.second;
	int left_count = countNode(left);
	int right_count = countNode(right);

[mamo3gr]

制約がわかりませんでしたが、root = nullptr のときここでクラッシュあるいは未定義動作しそうですね。