c语言求积水面积，整不会了啊

有几列由立方体垒叠的柱子并排放置，每列立方体的个数由数字表示。如01020200201，表示第一列无立方体，第二列有1个立方体，第三列无立方体，第四列有2个立方体叠成柱子，以此类推。则在第三列在雨后能积 1个单位面积 的水，第五列能积 2个单位面积 的水，第七、八列能积 4个单位面积，以此类推。最终统计积水面积总和。
输入：共两行
第一行：共输入了n列
第二行：n列立方体依次的数目(立方体堆叠个数不超过9)
输出：积水的面积
例：输入：5
11212
输出：1

#include<stdio.h>

int main()
{
    int n;
    int result = 0;

    printf("请输入有几列立方体：");
    scanf_s("%d", &n, 4);

    int* a = new int[n];
    int* b = new int[n];
    printf("分别输入每列立方体的个数:");
    for (int i = 0; i < n; ++i)
    {
        scanf_s("%d", &a[i], 4);
    }

    b[0] = a[0];
    for (int i = 1; i <= n-1; ++i)
    {
        if (a[i] < b[i - 1])
            b[i] = b[i - 1];
        else
            b[i] = a[i];
    }

    for (int i = 0; i < n; i++)
    {
        result += b[i] - a[i];
    }

    printf("%d", result);

    delete a;
    delete b;

    return 0;
}

这是我的理解哦，如果有错误请帮忙指出！

双指针可以解决，考虑只有0的地方可以积水

#define ll long long
 
class Solution {
public:
    long long maxWater(vector<int>& arr) {
        stack <int> st;
        ll ans = 0;
        for(int i = 0; i < arr.size(); ++i) {
            if(st.empty() || arr[i] < arr[ st.top() ]) {
                st.push(i);                                                                   //   (1) 维护一个单调递减栈
            }else {
                int h = i;
                while(!st.empty() && arr[i] >= arr[ st.top() ]) {            //   (2) 当前元素 >= 栈顶
                    if(st.size() == 1) {
                        st.pop();
                        h = i;                                                                 //   (3)  栈为 1 直接跳出
                        break;
                    }
                    h = st.top();
                    st.pop();
                    int dh = min(arr[i], arr[st.top()]) - arr[h];               // (4) 能够盛水的最大高度
                    int dw = i - h;                                                       // (5) 能够盛水的最大宽度
                    ans += (ll)dw * dh;                                               // (6) 相乘就是面积
                    arr[h] = arr[i];                                                       // (7) 这一步是为了削峰
                }
                st.push(h);
                 
            }
        }
        return ans;
    }
};

可以参考这篇文章：
https://blog.csdn.net/WhereIsHeroFrom/article/details/119074785 https://blog.csdn.net/WhereIsHeroFrom/article/details/119074785