积水问题

乌龟家的屋顶是凹凸不平的,所以每次雨后都会积水。为了知道屋顶是否会在暴雨后塌掉,他把屋顶的形 状给了你,希望你帮他计算暴雨后屋顶的积水总量。
乌龟的屋顶由顺次排在同一水平线上的 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;
}

猜你喜欢

转载自www.cnblogs.com/i-8023-/p/12091664.html