LeetCode - 42. Trapping Rain Water42. 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 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)

       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--;