竞赛讨论区 > 清空数组这一题,为什么我的写法不能通过所有用例呢

清空数组这一题,为什么我的写法不能通过所有用例呢

import sys

def foo(arr):
    if(len(arr) == 0): return 0
    arr = sorted(arr)
    cnt = 1
    prev = arr[0]
    for i in range(1, len(arr)):
        if(arr[i] == prev):
            cnt += 1
        elif(arr[i] == prev+1):
            prev = arr[i]
        else:
            cnt += 1
            prev = arr[i]
    return cnt

lines = sys.stdin.readlines()
arr = [int(x) for x in lines[1].strip().split()]
print(foo(arr))

想了很久,不知道是哪里有问题,还请各位指教?

另外头一次使用牛客,不知道是否可以查看测试不通过的用例是什么?

全部评论

(0) 回帖
加载中...
话题 回帖

等你来战

查看全部

热门推荐