美团 · 前缀和 · 算法编程题
美团 前缀和 n ≤ 200000 时限 2 秒 / 256 MB

题目描述

对于给定的由 n 个整数组成的数组 a_1,a_2,...,a_n ,计算其中有多少个三元组 (i,j,k) 满足 1≤ i < j < k ≤ n 且 a_i>a_k>a_j 。例如,在数组 \4,1,2,3 中三元组 (1,2,3) ,(1,2,4),(1,3,4) 都是满足条件的三元组。更具体地,计算:
Σ_1≤ i < j < k ≤ n[a_i>a_k>a_j]
请编写一个函数,计算并返回满足条件的三元组的数量。
【名词解释】
本题公式中的中括号代表艾弗森括号,具体地, [P] = cases 1 & 如果 P 为真 \0 & 如果 P 为假 cases 。

输入输出

输入描述
第一行输入一个整数 n (1≤ n ≤ 2× 10^5) 代表数组中的元素个数。
第二行输入 n 个整数 a_1,a_2,...,a_n (-10^9≤ a_i ≤ 10^9) 代表数组中的元素。
输出描述
输出一个整数,表示满足条件的三元组个数。

样例共 2 组

样例 1 · 在这个样例中,满足条件的三元组有: i=2 、 j=4 且 k=5 构成的三元组 \5,2,3 ; i=3 、 j=4 且 k=5 构成的三元组 \4,2,3 。
输入
5
1 5 4 2 3
输出
2
样例 2
输入
20
-6 -9 -90 -73 89 -90 2 19 52 -16 -41 -22 85 24 -22 66 75 78 48 -36
输出
134

算法解析依据充分

考点:数组 · 前缀和

数据规模 n ≤ 200000 | 限制 2 秒 / 256MB | 标准输入输出

题目画像

  • 数据规模:n ≤ 200000
  • 元素值域:a_i ≤ 1e9(注意整数类型选择,避免溢出)
  • 复杂度门槛:只允许 O(n) 或 O(n log n),且常数要小(注意读入效率)。
  • 源站时限:2 秒(牛客口径,非本题专属门槛)

解题思路

推荐方向:前缀和

预处理前缀和数组,把「区间求和」从 O(n) 降到 O(1)。

思路框架(前缀和 通法 · 非本题专属)

  1. 预处理 pre[i] = a[1]+...+a[i]。
  2. 区间 [l, r] 的和 = pre[r] - pre[l-1]。
  3. 若题目是「多次区间修改 + 最后统一查询」,改用差分数组:d[l]+=v, d[r+1]-=v,最后做一遍前缀和还原。
  4. 二维情况用二维前缀和(容斥原理)。

实现要点:开 long long,避免 n 较大时求和溢出。

复杂度:时间 预处理 O(n),每次查询 O(1) | 空间 O(n)

该范式的通法易错点

  • pre 数组下标没留出 0 号位导致 l-1 越界。
  • 求和结果超出 int 范围(1e5 个 1e9 相加会溢出 32 位整数)。

对照本题

  • 数据规模 n ≤ 200000,只允许 O(n) 或 O(n log n),且常数要小(注意读入效率)。
  • 元素值域最大到 1e9 —— 求和 / 相乘时记得开 64 位整数。

样例解读

样例 1:输入 5 / 1 5 4 2 3 → 输出 2

在这个样例中,满足条件的三元组有:

i=2 、 j=4 且 k=5 构成的三元组 \5,2,3 ;

i=3 、 j=4 且 k=5 构成的三元组 \4,2,3 。

解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。

本题来源:2025年秋招-美团-全栈岗-第二批笔试;2025年秋招-美团-算法策略端-第二批笔试;2025年秋招-美团-技术岗-第二批笔试。

‹ 上一题 全部编程题 下一题 ›
编程算法题为只读内容:无需作答,直接看题与解析 · 本站不提供在线判题 · 解析由校招宝本地引擎整理,非官方题解