summary-ranges.py

# https://leetcode.com/problems/summary-ranges/description/

class Solution:
    def summaryRanges(self, nums):
        """
        :type nums: List[int]
        :rtype: List[str]
        """
        output = []
        temp = ""

        for i in range(len(nums)):
            if len(temp) == 0:
                temp += str(nums[i])
                if i == len(nums) - 1:
                    output.append(temp)
                    temp = ""
                else:
                    if nums[i] + 1 != nums[i + 1]:
                        output.append(temp)
                        temp = ""

            else:
                if i == len(nums) - 1:
                    temp += "->"
                    temp += str(nums[i])
                    output.append(temp)
                    temp = ""
                else:
                    if nums[i] + 1 != nums[i + 1]:
                        temp += "->"
                        temp += str(nums[i])
                        output.append(temp)
                        temp = ""

        if temp:
            output.append(temp)
        return output


s = Solution()
print(s.summaryRanges([0, 1, 2, 4, 5, 7]))
print(s.summaryRanges([0, 2, 3, 4, 6, 8, 9]))