2019米哈游编程笔试--0万能数字

题目描述

给定一组数字（每个数字取值范围为0~100，可重复出现，其中0为万能数字，可以代表其他任何数字），判断它们是否是连续数字集合。
例如：
3 5 4 2 和3 5 0 2（让0代表4）和3 6 0 0 （一个0代表4，另外一个代表5）和3 0（让0代表2或4都可以）均为连续数字组合。而3 5 2和3 5 0 1和3 7 0 0和3 3 4 5（有重复的3）均不是连续数字集合。

N = input()
num = input()
n_zero = 0 //count 0
diff = [] //array 各项相减

arr = [int(n) for n in num.split()]
arr.sort()
idx = 0 //不是0 的第一个索引
for i in range(len(arr)):
    if arr[i] == 0:
        n_zero += 1
    else:
        idx = i
        break
        
for i in range(idx+1, len(arr)):
    diff.append(arr[i]-arr[i-1])

res = sum(diff) 
//判断是否全是0
if sum(arr) == 0:
    print("YES+"+str(len(arr)))
elif (res <= n_zero or res == len(arr)-1) and 0 not in diff:
    print('YES+'+str(n_zero))
else:
    print('NO+'+str(n_zero))