这道题目描述很清晰,直接两层循环,代码如下:

 class Solution(object):
def subarraysDivByK(self, A: 'List[int]', K: int) -> int:
n = len(A)
totalnum = 0
for i in range(n):
cursum = 0
for j in range(i,n):
cursum += A[j]
if cursum % K == 0:
totalnum += 1 return totalnum

显然,时间复杂度O(n^2),这样会超时。

因此需要优化,把时间复杂度减少O(n),一般这种类型的题目,都是要从数学角度分析,寻找规律。

先贴代码:

 class Solution(object):
def subarraysDivByK(self, A: 'List[int]', K: int) -> int:
n = len(A)
totalnum = 0
preDic = dict()
cursum = 0
for i in range(n):
cur = A[i]
cursum += cur
res = cursum % K
if res == 0:
totalnum += 1
if res in preDic:
totalnum += preDic[res]
preDic[res] += 1
else:
preDic[res] = 1
return totalnum

这里比较关键的代码就是13~17行,对于前m项的连加和,模K的余数,如果是第一次出现,则在字典中进行“标记”。

如果不是第一次出现,则执行第14行代码,这看起来很奇怪。

那就使用实际的示例,调试代码来分析,测试用例是[4,5,0,-2,-3,1],K值为5。

当i==0时,A[i]的值是4,(res==4)因此preDic中增加{4:1}这一项,

当i==1时,A[i]的值是5,(res==4)这说明从上一次出现res==4,到本次出现res==4之间的区间,可能出现了0到x个K,x的值是多少呢?

恰好就是preDic[res]的值!

假设preDic[4] = 2,那么2这个值一般解读为:“余数为4的次数”。

也可以解读为:“到目前为止,已经出现了几次K的倍数,并且余数为4”。而这个次数,是等于数值-1。

即preDic[4] == 1时,表示第一次出现余数4,那么之前就有0次K的倍数。

predDic[4] == 2时,表示第二次出现余数4,那么之前就有1次K的倍数。因此才会有第14行:totalnum += preDic[res]。

leetcode974的更多相关文章

  1. [Swift]LeetCode974. 和可被 K 整除的子数组 | Subarray Sums Divisible by K

    Given an array A of integers, return the number of (contiguous, non-empty) subarrays that have a sum ...

随机推荐

  1. 如何处理Excel空行问题

    在操作excel的时候, 可能会出现很多的无效数据行. 下面是一个我的简单处理方式 public static bool DataSetToExcel(DataSet dataSet, string ...

  2. 爬取网络图片到C盘存储的PermissionError: [Errno 13] Permission denied

    C盘根目录下不能拷进去文件,但可以新建文件夹的,“发生错误,操作被阻止,客户端没有所需的特权. 因为是系统目录,不要在里面拷贝文件,最好建立一个目录再放在里面. 硬要拷贝的话,可以使用管理员权限打开e ...

  3. Mysql基本代码操作

    Mysql的基本代码生成操作 创建一个数据库   (myschool是数据库名) create database myschool; 删除数据库 drop database myschool 创建一个 ...

  4. PowerShell 知识点总结

  5. nexus 数据库备份任务webhook 通知-另外一种方法

    使用benthos 做为webhook,是一种方法,功能很强大,但是有点复杂,所以换了一个更简单直接的webhook 工具 根据请求的数据,只处理关于db exporter 任务部分的消息,然后就是调 ...

  6. Yii2 设计模式——Yii2 中用到哪些设计模式?

    Yii 2 设计模式“包含了两个方面的内容:1. 设计模式,2. Yii 2 框架. <设计模式>一书虽然以JAVA语言来表达设计模式的思想,但是设计模式远不限制于某一种特定的语言,而是在 ...

  7. HanLP vs LTP 分词功能测试

    文章摘自github,本次测试选用 HanLP 1.6.0 , LTP 3.4.0 测试思路 使用同一份语料训练两个分词库,同一份测试数据测试两个分词库的性能. 语料库选取1998年01月的人民日报语 ...

  8. VisualSVN设置提交时必须输入日志信息

    VisualSVN设置提交时必须输入日志信息 1.svn提交时强制输入提交信息 为了阻止SVN提交空日志信息和垃圾文件可以在SVN服务器端强制必须填写日志信息,这时需用到pre-commit钩子脚本. ...

  9. sql语句可以截取指定字段后面的字符串

    select id,substring(Memo,charindex('数量',Memo)+3,len(Memo)-charindex('数量',Memo)) from trace where Mem ...

  10. Error in loadNamespace 的解决之道

    Error in loadNamespace(j <- i[[1L]], c(lib.loc, .libPaths()), versionCheck = vI[[j]])   在构建比较复杂的环 ...