华为机试---直方图内最大矩形

题目描述

有一个直方图，用一个整数数组表示，其中每列的宽度为1，求所给直方图包含的最大矩形面积。比如，对于直方图[2,7,9,4],它所包含的最大矩形的面积为14(即[7,9]包涵的7x2的矩形)。

给定一个直方图A及它的总宽度n，请返回最大矩形面积。保证直方图宽度小于等于500。保证结果在int范围内。

测试样例：

[2,7,9,4,1],5

返回：14

import java.util.Scanner;
 
public class Main {
public static void main(String[] args){
Scanner scan = new Scanner(System.in);
while(scan.hasNext()){
int N = scan.nextInt();
int[] A = new int[N];
for(int i = 0 ; i < N ; i++){
A[i] = scan.nextInt();
}
System.out.println(countArea(A , N));
}   
scan.close();
    }
public static int countArea(int[] A , int n){
int maxArea = 0;
int temp_int = 0;
int count = 1;//代表宽度
for(int i = 0 ; i < n ; i++){
temp_int = A[i];
count = 1;
//从当前元素向前和向后遍历计算当前高度下的最大宽度， 最大面积=当前高度*count
//1.向后，如果后面的高度（数组元素）大于等于当前的高度，count累加记录
for(int j = i + 1 ; j < n ; j++){
if(temp_int <= A[j]){
count++;
}else{
break;
}
}
//2.向前，如果前面的高度（数组元素）小于等于当前的高度，count累加记录
for(int k = i - 1 ; k >= 0 ; k--){
if(temp_int <= A[k]){
count++;
}else{
break;
}
}
   //3.此时已经找到当前高度下的最大宽度
maxArea = maxArea < (temp_int * count) ? (temp_int * count) : maxArea;
}
return maxArea;
}
}