last stone weight leetcode python solution