LeetCode - 42. Trapping Rain Water

42. Trapping Rain Water #

Problem's Link
#

 ----------------------------------------------------------------------------

Mean: 

在坐标上给你一些竖直放置的条形积木，问你这个积木能够容纳多少液体.

analyse:

首先找出最高的积木，然后从前往后一直扫到最高积木，从后往前一直扫到最高积木，两部分体积相加即可.

Time complexity: O(N)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-03-05-22.10
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long(LL);
typedef unsigned long long(ULL);
const double eps(1e-8);

class Solution
{
public:
   int trap(vector<int>& height)
   {
       int len=height.size(),front=0,res=0;
       int max_height=0,max_idx=len-1;
       for(int i=0;i<len;++i)
       {
           if(max_height<height[i])
           {
               max_height=height[i];
               max_idx=i;
           }
       }

       while(front<max_idx)
       {
           while(front+1<max_idx && height[front]<height[front+1])
               ++front;
           int stone=0;
           int back=front+1;
           while(back<max_idx && height[back]<height[front])
           {
               stone+=height[back];
               ++back;
           }
           res+=(back-front-1)*height[front]-stone;
           front=back;
       }

       int back=len-1;
       while(back>max_idx)
       {
           while(back-1>max_idx && height[back]<height[back-1])
               --back;
           int stone=0;
           int front=back-1;
           while(front>max_idx && height[front]<height[back])
           {
               stone+=height[front];
               --front;
           }
           res+=(back-front-1)*height[back]-stone;
           back=front;
       }
       return res;
   }
};

int main()
{
   freopen("H:\\Code_Fantasy\\in.txt","r",stdin);
   int n;
   while(cin>>n)
   {
       vector<int> ve;
       for(int i=0;i<n;++i)
       {
           int tempVal;
           cin>>tempVal;
           ve.push_back(tempVal);
       }
       Solution solution;
       cout<<solution.trap(ve)<<endl;
   }
   return 0;
}
/*

*/

 

2.方法二：时间O(n),空间(1)

class Solution
{
public:
   int trap(vector<int>& height)
   {
       int left=0,right=height.size()-1;
       int maxLeft=0,maxRight=0;
       int res=0;
       while(left<right)
       {
           if(height[left]<=height[right])
           {
               if(height[left]>maxLeft)
                   maxLeft=height[left];
               res+=maxLeft-height[left];
               left++;
           }
           else
           {
               if(height[right]>maxRight)
                   maxRight=height[right];
               res+=maxRight-height[right];
               right--;
           }
       }
       return res;
   }
};