• 作者:老汪软件技巧
  • 发表时间:2024-12-07 10:04
  • 浏览量:

在算法设计和优化中,前缀和算法是一种简单而强大的技术,它通过预处理数组数据来加速对数组子区间和的查询。

这种算法思想特别适用于需要频繁计算数组中连续子区间和的场景,如数据流问题、区间查询问题等。本文将详细介绍前缀和算法的思想、实现步骤以及其应用场景。

前缀和算法思想概述

前缀和算法的核心思想是预先计算并存储数组中每个位置之前所有元素的累积和,这样在需要计算任意子区间和时,可以直接通过查找前缀和数组中的特定元素来快速得出结果。

算法实现步骤1. 计算前缀和数组

前缀和数组的构建是算法的第一步。给定一个数组 A,长度为 n,我们创建一个新的数组 sum,其中 sum[i] 表示数组 A 中从第一个元素到第 i 个元素的累积和。具体计算方法如下:

这个步骤的时间复杂度为 (O(n)),其中 n 是数组 A 的长度。

2. 利用前缀和数组计算子区间和

一旦构建了前缀和数组,计算任意子区间 [i, j] 的和变得非常简单。子区间和可以通过以下公式快速计算:

这里需要注意的是,由于 sum[0] 存储的是数组的第一个元素,所以当 i = 0 时,公式变为 sum[j],因为 sum[0] 与 sum[i-1] 相等。

前缀和算法因其高效性,在多种算法问题中都有应用。以下是一些常见的应用场景:

动态数组的考虑

正如前文所述,前缀和算法在处理静态数组时非常有效。

然而,如果数组是动态变化的,即元素的值或位置会发生变化,那么可能需要定期重新计算前缀和数组,这会增加额外的计算开销。

在这种情况下,需要根据具体问题的特点来权衡使用前缀和算法的利弊。

算法题分割数组的最大差值描述

给定一个由若干整数组成的数组nums ,可以在数组内的任意位置进行分割,将该数组分割成两个非空子数组(即左数组和右数组),分别对子数组求和得到两个值,计算这两个值的差值,请输出所有分割方案中,差值最大的值。

输入描述

第一行输入数组中元素个数n,1 < n ≤ 100000

第二行输入数字序列,以空格进行分隔,数字取值为4字节整数

输出描述

输出差值的最大取值

题解

具体的步骤如下:

初始化变量:

遍历数组:

更新和计算:

输出结果:

import sys
# 读取输入的数组长度
length = int(sys.stdin.readline())
# 读取数字序列,并将其转换为整数列表
numbers = list(map(int, sys.stdin.readline().split()))
# 初始化前缀和数组,用于快速计算任意区间的和
prefixSum = [0] * length
prefixSum[0] = numbers[0]
for i in range(1, length):
    prefixSum[i] = prefixSum[i - 1] + numbers[i]
# 初始化最大差值变量
maxDifference = 0
# 遍历数组,计算每个分割点的左右数组和的差值
for i in range(length - 1):
    # 计算当前分割点左边数组的和
    leftSum = prefixSum[i]
    # 计算当前分割点右边数组的和
    rightSum = prefixSum[length - 1] - prefixSum[i]
    # 计算左右数组和的绝对差值
    difference = abs(leftSum - rightSum)
    # 更新最大差值
    maxDifference = max(maxDifference, difference)
# 输出最大差值

_【前缀和】算法思想,附两道道手撕题_【前缀和】算法思想,附两道道手撕题

print(maxDifference)

查找接口成功率最优时间段描述

服务之间交换的接口成功率作为服务调用关键质量特性,某个时间段内的接口失败率使用一个数组表示,

数组中每个元素都是单位时间内失败率数值,数组中的数值为0~100的整数,

给定一个数值(minAverageLost)表示某个时间段内平均失败率容忍值,即平均失败率小于等于minAverageLost,

找出数组中最长时间段,如果未找到则直接返回NULL。

输入描述

输入有两行内容,第一行为{minAverageLost},第二行为{数组},数组元素通过空格(” “)分隔,

minAverageLost及数组中元素取值范围为0~100的整数,数组元素的个数不会超过100个。

输出描述

找出平均值小于等于minAverageLost的最长时间段,输出数组下标对,格式{beginIndex}-{endIndx}(下标从0开始),

如果同时存在多个最长时间段,则输出多个下标对且下标对之间使用空格(” “)拼接,多个下标对按下标从小到大排序。

题解

解题思路如下:

数据读取:首先,我们需要从输入中获取两个关键参数:允许的平均失败率阈值以及记录失败率的数据数组。构建累积和数组:为了高效计算任意子区间的失败率总和,我们构建一个累积和数组。该数组的每个元素代表从数组起点到当前位置的失败率累加值。遍历子区间:接着,我们遍历所有可能的子区间,即所有可能的起始和结束索引组合。对于每个子区间,我们利用累积和数组快速确定该区间的失败率总和,并据此计算平均失败率。条件检查:对于每个子区间,我们验证其平均失败率是否不超过允许的阈值。如果满足条件,即记录该子区间。寻找最长子区间:在记录满足条件的子区间时,我们同时追踪最长的子区间长度。一旦发现更长的子区间,即更新最长长度,并重置结果列表,将新子区间加入。若发现等长的子区间,则直接添加到结果列表中。结果输出:最后,我们检查结果列表。若为空,则表示没有找到任何符合条件的子区间,输出"NULL"。否则,输出所有符合条件的子区间,若有多个,则按起始索引的升序排列。

通过这种方法,我们利用累积和数组高效地计算子区间失败率总和,并借助结果列表追踪所有满足条件的子区间,从而在单次遍历中找到所有符合条件的子区间,并快速确定最长的子区间。

# 读取输入的整数n,代表允许的平均失败率阈值
n = int(input())
# 读取输入的数字列表,并将其转换为整数列表
nums = list(map(int, input().split()))
# 获取列表的长度
m = len(nums)
# 初始化前缀和数组,长度为m+1,多一个0用于方便计算
preSum = [0] * (m + 1)
# 用于存储每个长度的子区间及其对应的索引范围
ans = {}
# 计算前缀和
for i in range(m):
    preSum[i + 1] = preSum[i] + nums[i]
# 使用滑动窗口优化算法遍历所有子区间
for i in range(m):
    for j in range(i, m):
        # 计算子区间的和
        subSum = preSum[j + 1] - preSum[i]  
        # 计算子区间的长度
        cnt = j - i + 1
        # 检查子区间的平均值是否小于等于允许的平均失败率阈值
        if subSum / cnt <= n:
            # 如果该长度的子区间已存在于ans字典中,则追加索引范围
            if cnt in ans:
                ans[cnt].append(f'{i}-{j}')
            # 否则,创建一个新的列表存储该长度的子区间索引范围
            else:
                ans[cnt] = [f'{i}-{j}']
# 找到最长的子区间长度
maxL = max(ans.keys())
# 输出最长子区间的所有索引范围,以空格分隔
print(' '.join(ans[maxL]))