leetCode 1(Python)