dotcpp 分苹果算法题分析

dotcpp题目:

时间限制: 3s 内存限制: 192MB 提交: 4117 解决: 1137
题目描述
小朋友排成一排，老师给他们分苹果。
小朋友从左到右标号1..N。有M个老师，每次第i个老师会给第Li个到第Ri个，一共Ri-Li+1个小朋友每人发Ci个苹果。
最后老师想知道每个小朋友有多少苹果。
输入格式
第一行两个整数N、M，表示小朋友个数和老师个数。 
接下来M行，每行三个整数Li、Ri、Ci，意义如题目表述。

数据规模和约定
100%的数据，N、M≤100  000，1≤Li≤Ri≤N，0≤Ci≤100。

输出格式
一行N个数，第i个数表示第i个小朋友手上的水果。 
样例输入
5 3
1 2 1
2 3 2
2 5 3
样例输出
1 6 5 3 3

时间超限但结果正确的解答:

# 读取小朋友和老师的数量
N, M = map(int, input().split())
# 创建一个长度为N+2的数组来存储每个小朋友的苹果数量
apples = [0] * (N + 2)
# 对于每个老师，我们将他们分发的苹果数量加到左边界，并从右边界的下一个位置减去
for _ in range(M):
    L, R, C = map(int, input().split())
    apples[L] += C
    apples[R + 1] -= C
# 遍历数组，将每个位置的苹果数量设置为它和前一个位置的苹果数量之和
for i in range(2, N + 1):
    apples[i] += apples[i - 1]
# 打印出每个小朋友的苹果数量
print(' '.join(map(str, apples[1:N+1])))

差分数组通过保存后一项减前一项省去了大量的遍历，有点像信号处理中的傅里叶变换，通过将数据转换到另一个维度就更容易计算

附测试简单暴力算法和差分数组性能差异测试代码

import io
import sys
import time
import random
from tqdm import tqdm

def generate_data(N, M):
    data = f"{N} {M}\n"
    for _ in tqdm(range(M), desc='creating testing data', unit='test sample'):
        L = random.randint(1, N)
        R = random.randint(L, N)
        C = random.randint(0, 100)
        data += f"{L} {R} {C}\n"
    return data

# 使用生成的数据
N = 10000  # 小朋友的数量
M = 5000   # 老师的数量
input_data = generate_data(N, M)

# 使用StringIO将输入数据重定向到stdin
sys.stdin = io.StringIO(input_data)

# 使用差分数组的方法
start = time.time()
N, M = map(int, input().split())
apples = [0] * (N + 2)
for _ in tqdm(range(M), desc='Processing teachers', unit='teachers'):
    L, R, C = map(int, input().split())
    apples[L] += C
    apples[R + 1] -= C

for i in range(2, N + 1):
    apples[i] += apples[i - 1]
#print(' '.join(map(str, apples[1:N+1])))
end = time.time()
print('差分数组的方法用时：', end - start, '秒')

# 重置输入数据
sys.stdin = io.StringIO(input_data)

# 使用双重循环的方法
start = time.time()
N, M = map(int, input().split())
apples = [0] * (N + 2)
for _ in tqdm(range(M), desc='Processing teachers', unit='teachers'):
    L, R, C = map(int, input().split())
    for i in range(L, R + 1):
        apples[i] += C
#print(' '.join(map(str, apples[1:N+1])))
end = time.time()
print('双重循环的方法用时：', end - start, '秒')