Update optimal_bst.cc