积水问题
乌龟家的屋顶是凹凸不平的,所以每次雨后都会积水。为了知道屋顶是否会在暴雨后塌掉,他把屋顶的形 状给了你,希望你帮他计算暴雨后屋顶的积水总量。
乌龟的屋顶由顺次排在同一水平线上的 n 个宽度为 1、高度为整数 (分别给出) 的瓦片组成。例如给定n = 5,瓦片的高度分别为 4, 2, 3, 5, 1,屋顶可以画在下图所示的网格中,灰色格子为瓦片。
暴雨过后,如果一个方格向左右两侧延伸都能到达瓦片占据的方格,它就会积水。所以图中波浪线格子在暴雨后会积水,屋顶的积水方格总数为 3。
输入
两个整数 n, R1,表示屋顶的宽度和生成数列的首项。从左向右数第 i (1<=i<=n)个瓦片的高度 ai = Ri mod 10
试题中使用的生成数列 R 定义如下:整数 0 ≤ R1 < 201701 在输入中给出。对于 i > 1,Ri = (Ri−1 × 6807 + 2831) mod 201701。
左右两边最高柱子中较小的那个和当前柱子比较
void init() {
cin >>n>>R1;
for (int i=0; i<n; ++i) {
height.push_back(R1%10);
R1 = (R1*6807 + 2831) % 201701;
}
}
int trap() {
int left = 0, right = n-1;//双指针
int ans = 0;
int l_max = height[0];
int r_max = height[n-1];
while (left <= right) {
l_max = max(l_max, height[left]);
r_max = max(r_max, height[right]);
// min(l_max,r_max) - height
if (l_max < r_max) {
ans += l_max - height[left];
++left;
} else {
ans += r_max - height[right];
--right;
}
}
return ans;
}