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

题目描述

有一个直方图，用一个整数数组表示，其中每列的宽度为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
;

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;

}

}