问题描述:
算法思想:使用递归方法来解决,每次计算出区间内的最小值,然后区间的所有元素都减去这个最小值,然后统计数值为零的数据对应的下标,然后再分别处理零之间的元素。
代码实现:
#include <cstdint>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<long long> h;
int solve(int l,int r){
if(l > r) return 0;
if(l == r){
if(h[l] == 0) return 0;
return 1;
}
long long minn = INT32_MAX;
// 找到区间内的最小值,区间内的全部元素都减去最小值
for(int i = l;i <= r;i++) minn = min(minn,h[i]);
for(int i = l;i <= r;i++) h[i] -= minn;
int ret(1);
vector<int> index;
// 记录为0的位置的下标
for(int i = l;i<=r;i++){
if(h[i] == 0) index.push_back(i);
}
// 保存左右边界区间
index.push_back(l-1);
index.push_back(r+1);
sort(index.begin(),index.end());
// 遍历每一个区间,对区间继续执行solve操作。
for(int i = 1;i<index.size();i++){
ret += solve(index[i-1]+1,index[i]-1);
}
return ret;
}
int main() {
int n;
cin >> n;
h.resize(n+1);
for (int i = 1; i <= n; i++) cin >> h[i];
cout << solve(1, n) << endl;
return 0;
}