15-4连续子串和的整除问题

问题描述

小M是一个五年级的小学生,今天他学习了整除的知识,想通过一些练习来巩固自己的理解。他写下了一个长度为 n 的正整数序列 a_0, a_1, ..., a_{n-1},然后想知道有多少个连续子序列的和能够被一个给定的正整数 b 整除。你能帮小M解决这个问题吗?


测试样例

样例1:

输入:n = 3,b = 3,sequence = [1, 2, 3]

输出:3

样例2:

输入:n = 4,b = 5,sequence = [5, 10, 15, 20]

输出:10

样例3:

输入:n = 5,b = 2,sequence = [1, 2, 3, 4, 5]

输出:6

python 复制代码
def solution(n, b, sequence):
    count = 0  # 用于记录符合条件的子序列数量
    
    # 外层循环,确定子序列的起始位置
    for start in range(n):
        current_sum = 0  # 当前子序列的和
        
        # 内层循环,确定子序列的结束位置
        for end in range(start, n):
            current_sum += sequence[end]  # 更新当前子序列的和
            
            # 检查当前和是否能被 b 整除
            if current_sum % b == 0:
                count += 1  # 如果可以整除,则计数加一
                
    return count  # 返回符合条件的子序列数量

if __name__ == "__main__":
    # 测试样例
    sequence1 = [1, 2, 3]
    print(solution(3, 3, sequence1) == 3)  # 输出: True

    sequence2 = [5, 10, 15, 20]
    print(solution(4, 5, sequence2) == 10)  # 输出: True

    sequence3 = [1, 2, 3, 4, 5]
    print(solution(5, 2, sequence3) == 6)  # 输出: True

    # 其他测试用例
    sequence4 = [1, 1, 1, 1]
    print(solution(4, 2, sequence4) == 10)  # 输出: True (所有子序列的和都是 1 或 2 的倍数)
相关推荐
太凉几秒前
Python之 sorted() 函数的基本语法
python
go54631584652 分钟前
修改Spatial-MLLM项目,使其专注于无人机航拍视频的空间理解
人工智能·算法·机器学习·架构·音视频·无人机
项目題供诗17 分钟前
黑马python(二十四)
开发语言·python
油泼辣子多加19 分钟前
【Torch】nn.BatchNorm1d算法详解
算法
nlog3n20 分钟前
基于 govaluate 的监控系统中,如何设计灵活可扩展的自定义表达式函数体系
算法·go
IT古董32 分钟前
【第三章:神经网络原理详解与Pytorch入门】01.神经网络算法理论详解与实践-(2)神经网络整体结构
pytorch·神经网络·算法
ThetaarSofVenice1 小时前
垃圾收集相关算法Test
java·jvm·算法
晓13131 小时前
OpenCV篇——项目(二)OCR文档扫描
人工智能·python·opencv·pycharm·ocr
小陈phd1 小时前
langchain从入门到精通(二十八)——RAG优化策略(六)集成多种检索器算法实现混合检索及问题转换总结
算法
是小王同学啊~1 小时前
(LangChain)RAG系统链路向量检索器之Retrievers(五)
python·算法·langchain