leetcode題解-python 645. 錯誤的集合 Set Mismatch (Easy)

645. 錯誤的集合 Set Mismatch (Easy)

集合 S 包含從1到 n 的整數。不幸的是，因爲數據錯誤，導致集合裏面某一個元素複製了成了集合裏面的另外一個元素的值，導致集合丟失了一個整數並且有一個元素重複。

給定一個數組 nums 代表了集合 S 發生錯誤後的結果。你的任務是首先尋找到重複出現的整數，再找到丟失的整數，將它們以數組的形式返回。

示例 1:

輸入: nums = [1,2,2,4]
輸出: [2,3]
注意:

給定數組的長度範圍是 [2, 10000]。
給定的數組是無序的。

github

鏈接: https://github.com/seattlegirl/leetcode/blob/master/set-mismatch.py.

題目代碼（python3）

思路：用時和內存消耗比較高，但是思路比較容易理解，用了兩個for循環，首先將字典的值全部初始化爲0（有沒有不用for循環初始化的好方法呀，求告知~~），再將集合中的每個數值作爲key值，數值出現的次數作爲value值。最終返回value=2的key值和value=0的key值。

#coding=utf-8
class Solution(object):
    def findErrorNums(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        dic={}
        for i in range(1,len(nums)+1):
            dic[i]=0
        for n in nums:
            dic[n]+=1
        a=list(dic.keys())[list(dic.values()).index(2)]
        b=list(dic.keys())[list(dic.values()).index(0)]
        return [a,b]
        
        
if __name__ == "__main__":
    print Solution().findErrorNums([1,2,2,4])