blob: 7185fc88b337d5b708920a248537e9d8b44b6eda (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
#!/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()
|