在二分查找结合后缀和(Prefix Sum / Suffix Sum)的问题中,哨兵值(Sentinel Value) 的作用是:
- 处理边界情况(如所有元素都不满足条件时)。
- 防止数组越界访问(如 sum[-1] 或 sum[n+1])。
- 简化代码逻辑(避免额外的 if 条件判断)。
什么是后缀和的哨兵值?
在计算后缀和时,通常我们会多定义一个位置(如 sum[n+1]),并初始化为 0:
sum[n+1] = 0; // 哨兵值
这样:
如果所有元素都不满足条件,sum[n+1] 会返回 0,避免越界。
在二分查找时,可以用 res = n+1 表示“无解”。
哨兵值的应用示例
问题描述
给定一个已排序数组 arr,支持两种操作:
全局加减 add(所有元素 arr[i] += change)。
查询 arr[i] + add > 0 的所有元素的和。
代码实现
#include <iostream>
#include <algorithm>
#define ll long long
using namespace std;
ll n, m, arr[500005], sum[500005], add = 0;
// 二分查找第一个 arr[i] + add > 0 的位置
ll findFirstPositive() {
ll left = 1, right = n, res = n + 1; // res初始化为n+1(哨兵值)
while (left <= right) {
ll mid = (left + right) / 2;
if (arr[mid] + add > 0) {
res = mid; // 记录可能的位置
right = mid - 1; // 继续往左找更小的
} else {
left = mid + 1; // 往右找
}
}
return res;
}
int main() {
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> arr[i];
sort(arr + 1, arr + n + 1); // 升序排序
// 计算后缀和,sum[n+1] = 0(哨兵值)
sum[n + 1] = 0;
for (int i = n; i >= 1; i--) {
sum[i] = sum[i + 1] + arr[i];
}
while (m--) {
int op;
cin >> op;
if (op == 1) {
ll change;
cin >> change;
add += change; // 全局增量
} else {
ll pos = findFirstPositive();
if (pos > n) {
cout << 0 << endl; // 所有arr[i]+add<=0,返回0
} else {
// 计算sum[pos] + (n - pos + 1) * add
cout << sum[pos] + (n - pos + 1) * add << endl;
}
}
}
return 0;
}
哨兵值的作用分析
(1) 防止越界
如果所有 arr[i] + add <= 0,findFirstPositive() 返回 n+1。
此时 sum[n+1] = 0,直接返回 0,不会访问非法内存。
(2) 简化逻辑
不需要额外判断 res 是否有效,直接用 if (pos > n) 处理无解情况。
(3) 正确计算
如果 pos 是合法下标(1 ≤ pos ≤ n),则:
sum[pos] 是 arr[pos] + arr[pos+1] + ... + arr[n]。
(n - pos + 1) * add 是增量 add 对这部分元素的贡献。
其他常见哨兵值
场景 | 哨兵值 | 作用 |
---|---|---|
前缀和 prefix[0] | prefix[0] = 0 | 计算区间和 sum[l..r] = prefix[r] - prefix[l-1] |
二分查找无解 | res = n+1 | 表示“没有找到” |
动态规划初始状态 | dp[0] = 1 | 处理空状态 |
关键总结
后缀和的哨兵值通常是 sum[n+1] = 0,用于处理“无解”情况。
二分查找时,res 初始化为 n+1,如果未更新则说明无解。
避免 sum[0] 的依赖,除非显式定义它(如 prefix[0] = 0)。
哨兵值的核心作用:防止越界 + 简化逻辑。
这样,代码更加健壮,能正确处理所有边界情况!