【C++】屋顶积水问题

题目大意：屋顶由若干根木桩组成，下雨后会产生积水，求最大积水量

输入：第一行一个整数N，表示屋顶由N根木桩组成。第二行N个整数，表示每根木桩的高度。

输出：一个整数，表示最大积水量

思路：初始将最左边的木桩设为lmax，最右边的木桩设为rmax，然后设置左右两个标志位i，j，哪边的max值小一些则哪边往对面走一格，此时如果该木桩更长则更新对应最大值，如果更短则表示能积水，则加上该木桩和此时对应最大值的差值。注意一定是要从小的max值往大的max值方向走，因为积水是根据短的那一边决定的。

#include<iostream>
#include<cstring>
using namespace std;
int l;
int h[1000];
int sum=0;

int main(void){
	cin>>l;
	int h[l];
	int i,j;
	for(i=0;i<l;i++){
		cin>>h[i];
	}
	int lmax=h[0],rmax=h[l-1];
	i=0;
	j=l-1;
	while(i<j){ //直到i,j相遇则结束 
		if(lmax<rmax){  //总是从小的那一边往大的那一边走一格 
			i++;
			if(h[i]>lmax) lmax=h[i];     //如果碰到更大的就更新最大值 
			else sum+=lmax-h[i];      //如果没有则加上差值 
		}
		else if(lmax>=rmax){
			j--;
			if(h[j]>rmax) rmax=h[j];
			else sum+=rmax-h[j];
		}
	}
	cout<<sum<<endl;
	return 0;
}