1. 程式人生 > >python 刷leetcode 題目(3)

python 刷leetcode 題目(3)

1、存在重複

給定一個整數陣列,判斷是否存在重複元素。

如果任何值在陣列中出現至少兩次,函式返回 true。如果陣列中每個元素都不相同,則返回 false。

示例 1:

輸入: [1,2,3,1]
輸出: true

示例 2:

輸入: [1,2,3,4]
輸出: false

示例 3:

輸入: [1,1,1,3,3,4,3,2,4,2]
輸出: true

程式碼如下:

class Solution:
    def containsDuplicate(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        nums.sort()
        for i in range(len(nums) -1):
            if nums[i] == nums[i+1]:
                return True
        return False

2、加一

給定一個非負整數組成的非空陣列,在該數的基礎上加一,返回一個新的陣列。

最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。

你可以假設除了整數 0 之外,這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入陣列表示數字 123。

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。

程式碼如下: 這個我採用兩種方法來做,大家可以根據自己的思路參考不同的做法。

class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        #### the first  method
        carry = 1
        for i in range(len(digits)-1, -1, -1):
            if digits[i] + carry == 10 :
                digits[i] = 0
                carry = 1
            else:
                digits[i] += carry
                carry = 0
        if carry == 1:
            digits.insert(0,1)
        return digits
       
        #### the second method 
#         if len(digits) == 0:
#             digits = [1]
#         elif digits[-1] == 9:
#             digits = self.plusOne(digits[:-1])
#             digits.extend([0])
#         else:
#             digits[-1] += 1
#         return digits
            
互相學習,如有錯誤,請指教。