题解 | 数独数组

def solve(nums):
    counts = [0]*9
    for i in nums:
        if i >9 or i < 1:
            return False
        if i == 1:
            counts[0]+=1
        if i == 2:
            counts[1]+=1
        if i == 3:
            counts[2]+=1
        if i == 4:
            counts[3]+=1
        if i == 5:
            counts[4]+=1
        if i == 6:
            counts[5]+=1
        if i == 7:
            counts[6]+=1
        if i == 8:
            counts[7]+=1
        if i == 9:
            counts[8]+=1
    m = n // 9
    d = n % 9
    if not all(x>=m for x in counts):
        return False
    if counts.count(m+1) != d:
        return False
    return True
n = int(input())
nums = list(map(int,input().split()))
if solve(nums):
    print("YES")
else:
    print("NO")