Leetcode If Else Problems