Modify balance_partitions fuction

"Only transactions producing least information gain from buckets over K
transactions will be re-distrbuted." I mis-understood this sentence.
This version still produce larger NPC than results in paper. But much closer.