python sol of q089
python sol of q089