溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

LeetCode如何返回某整數(shù)數(shù)組元素之和可被K整除的子數(shù)組數(shù)目

發(fā)布時間:2021-12-15 10:57:14 來源:億速云 閱讀:222 作者:小新 欄目:大數(shù)據(jù)

這篇文章主要為大家展示了“LeetCode如何返回某整數(shù)數(shù)組元素之和可被K整除的子數(shù)組數(shù)目”,內(nèi)容簡而易懂,條理清晰,希望能夠幫助大家解決疑惑,下面讓小編帶領(lǐng)大家一起研究并學(xué)習(xí)一下“LeetCode如何返回某整數(shù)數(shù)組元素之和可被K整除的子數(shù)組數(shù)目”這篇文章吧。


1

 題目描述

給定一個整數(shù)數(shù)組A,返回其中元素之和可被 K 整除的(連續(xù)、非空)子數(shù)組的數(shù)目。如輸入

A = [4,5,0,-2,-3,1], K = 5,返回7(因為有7個連續(xù)子數(shù)組的和可被5整除)。

2

 題解

思路:哈希表
本題跟LeetCode刷題DAY 17:和為k的子數(shù)組較為類似,定義pre(i)為[0,i]內(nèi)的所有元素和,則有pre(i)=pre(i-1)+A[i]關(guān)系,要找有多少個(pre(i)-pre(j-1))可被K整除。這里首先要介紹一下同余定理。
  • 同余定理設(shè)m是大于1的正整數(shù),a、b是整數(shù),如果m|(a-b),則稱a與b關(guān)于模m同余,記作a≡b(mod m)。

在本題中,即有(pre(i)-pre(j-1))|K等同于pre(i)≡pre(j-1)(mod K),因此我們在本題中可以建立哈希表,已余數(shù)為鍵,已該余數(shù)出現(xiàn)次數(shù)為值,計算哈希表中與pre(i)|K取值一樣的鍵對應(yīng)值的和即可。

class Solution:    def subarraysDivByK(self, A: List[int], K: int) -> int:        h_map = {0:1}        a = 0        ans = 0        for i in range(len(A)):            a += A[i]            if a%K in h_map:                ans += h_map[a%K]                h_map[a%K]+=1            else:                h_map[a%K]=1        return ans

以上是“LeetCode如何返回某整數(shù)數(shù)組元素之和可被K整除的子數(shù)組數(shù)目”這篇文章的所有內(nèi)容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內(nèi)容對大家有所幫助,如果還想學(xué)習(xí)更多知識,歡迎關(guān)注億速云行業(yè)資訊頻道!

向AI問一下細節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI