Trapping Rain Water

最新推荐文章于 2020-08-01 12:30:02 发布

真水壶

最新推荐文章于 2020-08-01 12:30:02 发布

阅读量413

点赞数

分类专栏： leetcode 文章标签： max

本文链接：https://blog.csdn.net/enjoyuseless/article/details/39343879

版权

leetcode 专栏收录该内容

154 篇文章 0 订阅

订阅专栏

//fun...
class Solution {
public:
    int trap(int A[], int n) {
		if(n<3)	return 0;
		vector<int> left_max(n,0),right_max(n,0);
		//left_max[i]: 从左到i,最大值的index
		//right_max[i]: 从右到i,最大值的index
		//from left to right
		int max_index=0;
		for(int i=0;i<n;i++){
			if(A[i]>A[max_index]){
				left_max[i]=i;
				max_index=i;
			}else{
				left_max[i]=max_index;
			}
		}
		//from right to left
		max_index=n-1;
		for(int i=n-1;i>=0;i--){
			if(A[i]>A[max_index]){
				right_max[i]=i;
				max_index=i;
			}else{
				right_max[i]=max_index;
			}
		}
		//get result
		int sum=0;
		for(int i=0;i<n;i++){
			if(left_max[i]!=i && right_max[i]!=i){
				sum=sum+min(A[left_max[i]],A[right_max[i]])-A[i];
			}
		}
		return sum;
    }
};