Python劃分數(shù)組為連續(xù)數(shù)字集合的練習(xí)

更新時間：2021年11月18日 09:20:59 作者：劉仕豪

這篇文章主要給大家分享的是Python劃分數(shù)組為連續(xù)數(shù)字集合的練習(xí)，下面文章首先對問題進行詳細描述，在根據(jù)問題提出解決方案，內(nèi)容詳細,需要的朋友可以參考一下，希望對你有所幫助

1、問題描述

給你一個整數(shù)數(shù)組 nums 和一個正整數(shù) k，請你判斷是否可以把這個數(shù)組劃分成一些由 k 個連續(xù)數(shù)字組成的集合。

如果可以，請返回 True；否則，返回 False。

示例 1：

輸入：nums = [1,2,3,3,4,4,5,6], k = 4

輸出：true

解釋：數(shù)組可以分成 [1,2,3,4] 和 [3,4,5,6]。

示例 2：

輸入：nums = [3,2,1,2,3,4,3,4,5,9,10,11], k = 3

輸出：true

解釋：數(shù)組可以分成 [1,2,3] , [2,3,4] , [3,4,5] 和 [9,10,11]。

示例 3：

輸入：nums = [3,3,2,2,1,1], k = 3

輸出：true

示例 4：

輸入：nums = [1,2,3,4], k = 3

輸出：false

解釋：數(shù)組不能分成幾個大小為 3 的子數(shù)組。

2、解決方案

剛剛拿到這道題，筆者想的是先找出數(shù)組中最小的一個數(shù)，然后根據(jù)k的值從數(shù)組中刪除相對應(yīng)的元素，比如k等于3，數(shù)組中最小數(shù)字為1，那么就從列表中刪除1，2，3三個元素，如果數(shù)組中沒有對應(yīng)的元素，那就該返回False。

如下題解：

def isPossibleDivide(nums, k):
     nums = sorted(nums)
     for _ in range(len(nums)//k):
         minv = nums[0]
         for _ in range(k):
             if minv in nums:
                 nums.remove(a)
                 minv +=1
     return len(nums) == 0

但是在第二個for循環(huán)里面有過多操作，如果k的值太大，那么代碼運行內(nèi)存便會很大，在規(guī)定內(nèi)存內(nèi)運行便會超時。于是筆者想到了第二種方法，雖然代碼量大一點，但是相對于第一種，時間復(fù)雜度更小，不容易超時，用集合找出數(shù)組中出現(xiàn)過的數(shù)字，再用字典統(tǒng)計每個數(shù)字出現(xiàn)的次數(shù)，設(shè)置判定條件，再根據(jù)連續(xù)判定條件返回對應(yīng)布爾型。

python代碼：

def isPossibleDivide(nums, k):
     n = len(nums)
     if n % k != 0:
         return False
     # 用集合記錄可能的數(shù)字
     s = set(nums)
     minList = list(s)
     minList.sort()
     # 用字典存儲每個數(shù)字出現(xiàn)的次數(shù)
     d = dict()
     for num in nums:
         if num not in d:
             d[num] = 0
         d[num] += 1
     # 判斷每組是否可由k個連續(xù)數(shù)字構(gòu)成
     m = n // k  # m組
     start = 0  # 起始位置
     for mi in range(m):
         if start >= len(minList):
             return False
         minv = minList[start]
         flag = True
         t = start
         for key in range(minv, minv +  k):
             if key not in d:
                 return False
             if d[key] < 1:
                 return False
             elif d[key] == 1:
                 d[key] -= 1
                 t += 1
             elif d[key] > 1:
                 d[key] -= 1
                 if flag:
                     start = t
                     flag = False
         if flag:
             start = t
     return True

3、結(jié)語

在遇到這類編程題時，要運用多種方法嘗試求解，考慮時間復(fù)雜度和空間復(fù)雜度等多方面因素尋找最優(yōu)解法。

到此這篇關(guān)于Python劃分數(shù)組為連續(xù)數(shù)字集合的練習(xí)的文章就介紹到這了,更多相關(guān)Python劃分數(shù)組為連續(xù)數(shù)字集合內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: