blob: 7185fc88b337d5b708920a248537e9d8b44b6eda (
plain) (
tree)
|
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
class Solution:
def findErrorNums(self, nums):
counts = {}
one = 0
for num in nums:
if num in counts:
counts[num] += 1
one = num
else:
counts[num] = 1
total = len(nums) * (len(nums) + 1) // 2
return [one, total - sum(nums) + one]
def main():
solution = Solution()
print(solution.findErrorNums([1, 2, 2, 4]))
print(solution.findErrorNums([1, 1]))
if __name__ == "__main__":
main()
|
