Update heap_algo.go