leetcode 2965. Find Missing and Repeated Values

https://leetcode.com/problems/find-missing-and-repeated-values/description/

Find Missing and Repeated Values - LeetCode

 

class Solution:
    def findMissingAndRepeatedValues(self, grid: List[List[int]]) -> List[int]:
        check = [0] * (len(grid)**2 + 1)
        for i in grid:
            for j in i:
                check[j] += 1
        answer = [0, 0]
 
        for i, v in enumerate(check):
            if v == 0:
                answer[1] = i
            if v > 1:
                answer[0] = i
        
        return answer