某互联网公司的题

import sys

if __name__ == '__main__':
    for x in sys.stdin:
        a = x.strip().split(',')
        # b = [1, 2, 3, 4, 6, 8, 9, 10]
        b = list(map(int, a))
        b.append(b[-1] + 2)  # 确保边界是自己想要的边界
        result = []
        start = 0
        end = 0
        for i in range(len(b) - 1):
            if b[i + 1] == b[i] + 1:
                end += 1
            else:
                if start != end:
                    result.append("-".join([str(b[start]), str(b[end])]))
                else:
                    result.append(b[start])
                end = start = i + 1
        print(result)
1,2,3,4,6,8,9,10
['1-4', 6, '8-10']

猜你喜欢

转载自blog.csdn.net/TomorrowAndTuture/article/details/105113564