You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

整数列表s长度重叠切片的按列求和技术实现问询

Hey 伙计,我来帮你把这个问题理清楚,再给你两种实用的实现方案,保证易懂好上手!

问题明确

咱们的目标很清晰:给定整数列表L和整数s,要计算L中所有长度为s的可重叠连续切片的按列求和结果,并且这个结果列表要和原列表L保持位置对应关系。

先搞懂核心概念

什么是s长度可重叠切片?

就是从原列表里取所有连续的、长度刚好是s的子序列,允许切片之间重叠。比如L = [1,2,3,4]s=2时,切片就是[1,2][2,3][3,4]——后一个切片和前一个共享了一个元素,这就是“可重叠”的含义。

怎么保持相对位置?

把这些切片排成阶梯状就一目了然了:第一个切片对齐原列表的开头,第二个切片往后偏移1位,第三个再偏移1位……像这样:

1 2
  2 3
    3 4

按列求和的话,每一列的结果就对应原列表中对应位置的元素被所有包含它的切片累加后的总和。比如原列表索引1的元素2,在第一个和第二个切片里都出现了,所以求和就是2+2=4

规格限制要注意

  • s必须满足:1 < s < len(L)(大于1,且严格小于原列表的长度)
  • L至少要有3个元素(毕竟s得小于它的长度,还得大于1嘛)

实现方案1:基础双重循环版

思路非常直接:先找出所有符合要求的切片,再把每个切片里的元素加到结果列表对应的位置上。

def column_sum_overlapping_slices(L, s):
    # 初始化结果列表,长度和原列表一致,初始值全为0
    result = [0] * len(L)
    # 遍历所有切片的起始索引(从0到len(L)-s,保证切片长度刚好是s)
    for start_idx in range(len(L) - s + 1):
        # 遍历当前切片里的每个元素
        for offset in range(s):
            # 原列表中对应的位置是start_idx + offset,把元素值累加进去
            result[start_idx + offset] += L[start_idx + offset]
    return result

# 测试示例
if __name__ == "__main__":
    # 示例1
    L1 = [1, 2, 3, 4]
    s1 = 2
    print(column_sum_overlapping_slices(L1, s1))  # 输出: [1, 4, 6, 4]
    
    # 示例2
    L2 = [5, 1, 3, 2, 4]
    s2 = 3
    print(column_sum_overlapping_slices(L2, s2))  # 输出: [5, 2, 9, 4, 4]

这个方法逻辑简单,容易理解,但时间复杂度是O(n*s),如果原列表特别长、s也很大的话,效率会稍低一些。

实现方案2:优化版(O(n)时间复杂度)

其实咱们可以换个思路:按列求和本质上是原列表中每个元素被多少个切片包含,然后用元素值乘以这个次数。我们只需要算出每个元素的被包含次数就行:

  • 原列表前s-1个元素(索引0s-2):第j个元素被包含的次数是j+1(比如索引0的元素只在第一个切片里,索引1的元素在第1、2个切片里,以此类推)
  • 中间的元素(索引s-1len(L)-s):每个元素都会被s个切片包含,次数就是s
  • 最后s-1个元素(索引len(L)-s+1len(L)-1):第j个元素被包含的次数是len(L)-j(比如最后一个元素只在最后一个切片里)

基于这个思路,代码就变成了:

def column_sum_overlapping_slices_optimized(L, s):
    n = len(L)
    result = []
    for j in range(n):
        if j < s:
            # 前s个元素,次数是j+1
            count = j + 1
        elif j > n - s - 1:
            # 后s个元素,次数是n-j
            count = n - j
        else:
            # 中间元素,次数是s
            count = s
        result.append(L[j] * count)
    return result

# 测试同样的示例
if __name__ == "__main__":
    L1 = [1, 2, 3, 4]
    s1 = 2
    print(column_sum_overlapping_slices_optimized(L1, s1))  # 输出: [1, 4, 6, 4]
    
    L2 = [5, 1, 3, 2, 4]
    s2 = 3
    print(column_sum_overlapping_slices_optimized(L2, s2))  # 输出: [5, 2, 9, 4, 4]

这个方法只需要遍历一次原列表,时间复杂度是O(n),效率高很多,适合处理大规模数据。

内容的提问来源于stack exchange,提问作者Mr. Xcoder

火山引擎 最新活动