题目:给定一个数组,打印出每一个元素的下一个更大的元素( Next Greater Element,NGE )。一个元素x的下一个更大的元素是指在x的右边第一个比该元素大的元素。如果没有更大的元素,则输出-1。
方法1 枚举
使用二重循环,分别遍历所有的元素,即可找到该元素的NGE。时间复杂度O(n^2).
方法2 使用栈
可以用栈来维护一个递减的序列,里面存储的数组左边未找到NGE的元素,初始时只包含第一个元素。我们可以假定栈顶就是最小的元素,因此可以用从栈顶top开始和后面的元素next比较。如果top<next则说明,找到了top的NGE,可以弹出栈。然后继续比较栈顶元素top和next,直到栈为空或 next <= top。然后把next加入栈,以便查找next的NGE。由于只有一次遍历,时间复杂度为O(n)*
方法2 的实现代码如下:
/**
* @Title: NextGreaterElement.java
* @Package stack
* @Description: TODO
* @author peidong
* @date 2017-5-4 上午9:20:17
* @version V1.0
*/
package stack;
import java.util.Stack;
public class NextGreaterElement {
public static void findNGE(int[] arr, int len){
//初始化栈
Stack<Integer> stack = new Stack<Integer>();
//将数组首个元素入栈
stack.push(arr[0]);
int top, next; //定义两个指针变量
for(int i = 1; i < len ; i++){
next = arr[i];
top = stack.peek();
//判断是否找到栈顶元素的NGE,此处是循环判断栈中元素与当前元素next值得比较,输出不会按照原数组顺序
while((!stack.isEmpty()) && top < next){
System.out.printf("%d --> %d", top, next);
System.out.println();
stack.pop(); //出栈
//继续判断栈顶
if(!stack.isEmpty()){
top = stack.peek();
}
}
//将下一个元素入栈
stack.push(next);
}
while(!stack.isEmpty()){
top = stack.peek();
stack.pop();
System.out.printf("%d --> %d", top, -1);
System.out.println();
}
}
/**
* @Title: main
* @Description: TODO
* @param @param args
* @return void
* @throws
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
int arr[]= {11, 13, 10, 5, 12,21, 3};
int len = arr.length;
findNGE(arr, len);
}
}