結果 : if else in python hackerrank solution