程序员代码面试指南刷题--第八章.容器盛水问题

题目描述

给定一个整形数组arr，已知其中所有的值都是非负的，将这个数组看作一个容器，请返回容器能装多少水。
具体请参考样例解释

输入描述:

第一行一个整数N，表示数组长度。

接下来一行N个数表示数组内的数。

输出描述:

输出一个整数表示能装多少水。

示例1
输入
6
3 1 2 5 2 4

输出
5



示例2
输入
5
4 5 1 3 2

输出
2

解法一：暴力（这法我都不会…）

import java.io.*;
public class Main{
    public static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    public static void main(String[] args) throws IOException{
        int len = Integer.parseInt(br.readLine());
        String[] ss = br.readLine().trim().split(" ");
        int[] a = new int[len];
        for(int i=0;i<len;i++){
            a[i] = Integer.parseInt(ss[i]);
        }
        int result = getVolume(a);
        System.out.print(result);
    }
    public static int getVolume(int[] a){
        if(a==null||a.length<3) return 0;
        int volume = 0;
        for(int i=1;i<a.length-1;i++){
            int leftmax = 0;
            int rightmax = 0;
            for(int j=0;j<i;j++){
                leftmax = leftmax<a[j]?a[j]:leftmax;
            }
            for(int k=i+1;k<a.length;k++){
                rightmax = rightmax<a[k]?a[k]:rightmax;
            }
            volume += Math.max(Math.min(leftmax,rightmax)-a[i],0); 
        }
        return volume;
    }
}

解法二：空间换时间

思路： 先构建左右最大值数组(这里用long防止溢出）

import java.io.*;
public class Main{
    public static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    public static void main(String[] args) throws IOException{
        int len = Integer.parseInt(br.readLine());
        String[] ss = br.readLine().trim().split(" ");
        long[] a = new long[len];
        for(int i=0;i<len;i++){
            a[i] = Long.parseLong(ss[i]);
        }
        long result = getVolume(a);
        System.out.print(result);
    }
    public static long getVolume(long[] a){
        if(a==null||a.length<3) return 0;
        long volume = 0;
        long[] leftmax = new long[a.length];  
        long[] rightmax = new long[a.length];
        leftmax[0] = a[0];
        rightmax[a.length-1] = a[a.length-1];
        for(int i=1;i<a.length;i++){
            leftmax[i] = Math.max(leftmax[i-1],a[i]);
        }
        for(int i=a.length-2;i>=0;i--){
            rightmax[i] = Math.max(rightmax[i+1],a[i]);
        }
        for(int i=1;i<a.length-1;i++){
            volume += Math.max(Math.min(leftmax[i-1],rightmax[i+1])-a[i],0);
        }
        return volume;
    }
}

解法三：牛皮

import java.io.*;
public class Main{
    public static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    public static void main(String[] args) throws IOException{
        int len = Integer.parseInt(br.readLine());
        String[] ss = br.readLine().trim().split(" ");
        long[] a = new long[len];
        for(int i=0;i<len;i++){
            a[i] = Long.parseLong(ss[i]);
        }
        long result = getVolume(a);
        System.out.print(result);
    }
    public static long getVolume(long[] a){
        if(a==null||a.length<3) return 0;
        long leftmax = a[0];
        long rightmax = a[a.length-1];
        int l = 1;
        int r = a.length-2;
        long volume = 0;
        while(l<=r){
            if(leftmax<=rightmax){
                volume += Math.max(leftmax-a[l],0);
                leftmax = Math.max(leftmax,a[l++]);
            }else{
                volume += Math.max(rightmax-a[r],0);
                rightmax = Math.max(rightmax,a[r--]);
            }
        }
        return volume;
    }
}