Fixed util function and AC
Fixed util function and AC

shichao-an/leetcode-python,shichao-an/leetcode-python,quietcoolwu/leetcode-python,huanqi/leetcode-python,quietcoolwu/leetcode-python,huanqi/leetcode-python