Python多維列表中的坑如何解決

這篇文章主要講解了“Python多維列表中的坑如何解決”，文中的講解內(nèi)容簡單清晰，易于學(xué)習(xí)與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學(xué)習(xí)“Python多維列表中的坑如何解決”吧！

數(shù)組常用想法總結(jié)：

（以下默認(rèn)nums為數(shù)組。） 1.遍歷數(shù)組 遍歷：

for num in nums:
	xxxx

帶索引遍歷

for idx,num in enumerate(nums):
	xxxx

2.動態(tài)規(guī)劃(dp) 動態(tài)規(guī)劃一般可以用一個數(shù)組保存狀態(tài)。見53.最大子數(shù)組和。 用數(shù)組保存狀態(tài)是非常常用的做法。例如36.有效的數(shù)獨(dú)、 73. 矩陣置零。

3.雙指針 見88.合并兩個有序數(shù)組、350.兩個數(shù)組的交集 II可以是左右指針對一個數(shù)組使用。 也可以是兩個指針遍歷兩個數(shù)組。while index1<m and index2<n:

列表常用函數(shù)

Python中一般用list實(shí)現(xiàn)可變數(shù)組。 下面是list常用的函數(shù)。 （可變序列類型通用操作，只有.sort是list獨(dú)有的。參考序列操作文檔）

多維列表的一個坑

創(chuàng)建多維列表，一般用

w, h = 2, 3
A = [[None] * w for i in range(h)]

等價(jià)于

A = [None] * 3
for i in range(3):
    A[i] = [None] * 2

而不是

 A = [[None] * 2] * 3

原因在于用*對列表執(zhí)行重復(fù)操作并不會創(chuàng)建副本，而只是創(chuàng)建現(xiàn)有對象的引用。 *3創(chuàng)建的是包含 3 個引用的列表，每個引用指向的是同一個長度為 2 的列表。 如果你給一項(xiàng)賦值，就會發(fā)現(xiàn)這個問題：

>>> A[0][0] = 5
>>> A
[[5, None], [5, None], [5, None]]

第1天

217. 存在重復(fù)元素

給定數(shù)組，判斷是否存在重復(fù)元素。 做法：

1. 直接遍歷（窮舉）

2. 排序后，比較每個元素和下一個元素

3. 哈希表

直接遍歷會超時。 2的時間復(fù)雜度是O(nlogn) 也就是排序的時間復(fù)雜度 3的時間復(fù)雜度是O(n)，但需要額外的O(n)輔助空間。 （窮舉法基本都能想到，但很容易超時，后面只有在窮舉法能通過時才列出來。）

3比較簡單，這里寫一下3的做法：

return len(nums) != len(set(nums))

53. 最大子數(shù)組和

給定數(shù)組，求其中一個連續(xù)數(shù)組和的最大值。

比較容易想到的是用一個數(shù)組記錄目前位置最大的值（動態(tài)規(guī)劃）。

用 dp[i] 表示以i位置結(jié)尾的連續(xù)數(shù)組和的最大值。 最后返回dp數(shù)組中最大值。

class Solution:
    def maxSubArray(self, nums: List[int]) -> int:
        length = len(nums)
        dp = [0 for i in range(length)]
        for i in range(length):
            dp[i] = max(dp[i - 1], 0) + nums[i]
        return max(dp)

題解給出了一種省略dp數(shù)組的方法：

class Solution:
    def maxSubArray(self, nums: List[int]) -> int:
        pre = 0
        res = nums[0]
        for x in nums:
            pre = max(pre+x ,x)
            res = max(res, pre)
        return res

第2天

1. 兩數(shù)之和

找出數(shù)組中兩個數(shù)之和等于target的兩數(shù)下標(biāo)。

暴力枚舉可以

但時間較長，時間復(fù)雜度$O(N^2)$

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        n = len(nums)
        for i in range(n):
            for j in range(i + 1, n):
                if nums[i] + nums[j] == target:
                    return [i, j]
        
        return []

哈希表

官方題解的一個比較巧妙的方式：使用哈希表（字典） 用字典記錄出現(xiàn)過的數(shù)字的位置。 時間復(fù)雜度$O(N)$，空間復(fù)雜度$O(N)$

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        hashtable = dict()
        for i, num in enumerate(nums):
            if target - num in hashtable:
                return [hashtable[target - num], i]
            hashtable[nums[i]] = i
        return []

88. 合并兩個有序數(shù)組

兩個有序數(shù)組，將第二個數(shù)組nums2合并到第一個數(shù)組nums1。

雙指針

1.可以用雙指針遍歷兩個數(shù)組：

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        # 兩個中存在空數(shù)組的時，直接返回
        if m == 0:
            nums1[:] = nums2[:]
            return
        if n == 0:
            return

        index1,index2 = 0,0
        t = []
        while index1<m and index2<n:
            if nums1[index1] <= nums2[index2]:
                t.append(nums1[index1])
                index1 += 1
            else:
                t.append(nums2[index2])
                index2 += 1 
        
        if index1 < m:
            t += nums1[index1:m]
        else:
            t += nums2[index2:n]

        nums1[:] = t[:]

官方版本，更簡潔、清楚。

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        sorted = []
        p1, p2 = 0, 0
        while p1 < m or p2 < n:
            if p1 == m:
                sorted.append(nums2[p2])
                p2 += 1
            elif p2 == n:
                sorted.append(nums1[p1])
                p1 += 1
            elif nums1[p1] < nums2[p2]:
                sorted.append(nums1[p1])
                p1 += 1
            else:
                sorted.append(nums2[p2])
                p2 += 1
        nums1[:] = sorted

（暴力） 追加后排序

2. 更簡單粗暴的方式是直接將nums2追加到nums1后，進(jìn)行排序。 及其簡單而且效果很好。

class Solution:
    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        """
        nums1[m:] = nums2
        nums1.sort()

第3天

350. 兩個數(shù)組的交集 II

以數(shù)組形式返回兩數(shù)組的交集（數(shù)組形式，返回結(jié)果中每個元素出現(xiàn)的次數(shù)，應(yīng)與元素在兩個數(shù)組中都出現(xiàn)的次數(shù)一致）。 排序后雙指針遍歷。

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        nums1.sort()
        nums2.sort()
        i = 0
        j = 0
        result = []
        while i<len(nums1) and j<len(nums2):
            if(nums1[i]<nums2[j]):
                i+=1
            elif(nums1[i]>nums2[j]):
                j+=1
            else:
                result.append(nums1[i])
                i+=1
                j+=1
       
        return  result

121. 買賣股票的最佳時機(jī)

只需要記錄下當(dāng)前最低價(jià)，遍歷價(jià)格過程中，用當(dāng)前價(jià)格-最低價(jià) 就是當(dāng)前可獲得的最大利潤。另外如果出現(xiàn)了更低的價(jià)格，則最低價(jià)也要更新。（一個樸素的想法，要是我在最低點(diǎn)買進(jìn)就好了） 總的最大利潤就是這些利潤中的最大值。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        r = 0
        min_price = float('inf')  # float('inf')表示正無窮
        for price in prices:
            min_price = min(min_price, price)  # 截止到當(dāng)前的最低價(jià)（買入價(jià)）
            r = max(r, price - min_price)  # 截止到目前的最高利潤
        return r

第4天

566. 重塑矩陣

給定一個mxn的數(shù)組，重構(gòu)為rxc的數(shù)組。 比較簡單的想法是把數(shù)組拉平為一位數(shù)組，然后逐個填充到新的數(shù)組中：

class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        m,n = len(mat), len(mat[0])
        if m*n != r*c:
            return mat
        arr = []
        for row in mat:
            for x in row:
                arr.append(x)
        arr_index = 0
        newmat = [[0 for j in range(c)]for i in range(r)]
        for i in range(r):
            for j in range(c):
                newmat[i][j] = arr[arr_index]
                arr_index += 1
        return newmat

官方提供了一種直接計(jì)算下標(biāo)的方法：

class Solution:
    def matrixReshape(self, nums: List[List[int]], r: int, c: int) -> List[List[int]]:
        m, n = len(nums), len(nums[0])
        if m * n != r * c:
            return nums
        
        ans = [[0] * c for _ in range(r)]
        for x in range(m * n):
            ans[x // c][x % c] = nums[x // n][x % n]
        
        return ans

118. 楊輝三角

找規(guī)律題?？梢灾苯影凑丈傻囊?guī)律生成數(shù)組。在「楊輝三角」中，每個數(shù)是它左上方和右上方的數(shù)的和。

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        res = [[]for _ in range(numRows)]
        res[0] = [1]
        for i in range(1,numRows):
            res[i].append(1)
            for j in range(0,len(res[i-1])-1):
                res[i].append(res[i-1][j] + res[i-1][j+1])
            res[i].append(1)

        return res

第5天

36. 有效的數(shù)獨(dú)

判斷當(dāng)前數(shù)獨(dú)是否有效（不需要填充數(shù)獨(dú)） 只要用3個二維數(shù)組維護(hù)9行、9列、9個九宮格。

class Solution:
    def isValidSudoku(self, board: List[List[str]]) -> bool:
        row = [[] * 9 for _ in range(9)]
        col = [[] * 9 for _ in range(9)]
        nine = [[] * 9 for _ in range(9)]
        for i in range(len(board)):
            for j in range(len(board[0])):
                tmp = board[i][j]
                if not tmp.isdigit():
                    continue
                if (tmp in row[i]) or (tmp in col[j]) or (tmp in nine[(j // 3) * 3 + (i // 3)]):
                    return False
                row[i].append(tmp)
                col[j].append(tmp)
                nine[(j // 3) * 3 + (i // 3)].append(tmp)
        return True

73. 矩陣置零

如果一個元素為 0 ，則將其所在行和列的所有元素都設(shè)為 0 。請使用 原地 算法。 A： 利用數(shù)組的首行和首列來記錄 0 值 另外用兩個布爾值記錄首行首列是否需要置0

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        #標(biāo)記
        m,n = len(matrix), len(matrix[0])
        row = any(x == 0 for x in matrix[0])
        col = any(matrix[r][0] == 0 for r in range(m) )
        
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == 0:
                    matrix[i][0] = 0
                    matrix[0][j] = 0
                    
        #置零
        for i in range(1,m):
            for j in range(1,n):
                if matrix[i][0] == 0 or matrix[0][j] == 0:
                    matrix[i][j] = 0
        if row:
            for j in range(0,n):
                matrix[0][j] = 0
        if col:
            for i in range(0,m):
                matrix[i][0] = 0

感謝各位的閱讀，以上就是“Python多維列表中的坑如何解決”的內(nèi)容了，經(jīng)過本文的學(xué)習(xí)后，相信大家對Python多維列表中的坑如何解決這一問題有了更深刻的體會，具體使用情況還需要大家實(shí)踐驗(yàn)證。這里是億速云，小編將為大家推送更多相關(guān)知識點(diǎn)的文章，歡迎關(guān)注！