P1007：小核桃数组变换难题，Python代码揭秘最终平方和！

以下是解决该问题的Python代码： ```python MOD = 109 + 7 # 读取输入 n, k = map(int, input().split()) # 构建初始数组 arr = [] for _ in range(k): a, b = map(int, input().split()) arr.extend([b] * a) # 进行数组变换 while len(arr) > 2: new_arr = [] for i in range(0, len(arr), 2): if i + 1 < len(arr): new_arr.append(arr[i] + arr[i + 1]) else: new_arr.append(arr[i]) arr = new_arr # 计算最后的两个数字的平方和并取模 result = (arr[0]2 + arr[1]2) % MOD print(result) ``` 代码解释 1. 读取输入：首先读取数组的长度`n`和信息的条数`k`，然后根据每条信息构建初始数组`arr`。 2. 数组变换：使用一个`while`循环，只要数组的长度大于2，就继续进行变换。在每次变换中，将相邻的两个数字相加，构建新的数组`new_arr`。 3. 计算平方和并取模：当数组的长度变为2时，计算这两个数字的平方和，并对`MOD`取模。 4. 输出结果：输出最后的结果。 复杂度分析 - 时间复杂度：每次变换数组的长度至少减半，因此变换的次数为$O(log n)$。每次变换需要遍历数组，时间复杂度为$O(n)$。总的时间复杂度为$O(n log n)$。 - 空间复杂度：主要的空间开销是存储数组，因此空间复杂度为$O(n)$。 ######[AI问答 | 416点数解答 | 2026-02-07 18:24:18]