题目解析
每个人会击中前面第一个比自己高的人,如上图所示。
用栈来存储数组的递减的子序列,从而找到每一个被击中的人。
遍历数组时,
- 若栈顶元素大于当前数组值,则栈顶元素被击中,将数字值入栈
- 若栈顶元素值小于或等于数组值,则将栈顶元素弹出,判断新的栈顶值大小。若栈为空,说明没有比当前值大的,将当前数数组值入栈。
JAVA代码
public long solve (int n, int[] a) { // write code here if(n <= 1){ return 0; } long ans = 0; Stack<Integer> stack = new Stack<>(); // i 表示第i个人 for(int i = 1;i <= n;i++){ while(!stack.isEmpty() && a[i - 1] >= a[stack.peek() - 1]){ stack.pop(); } if(!stack.isEmpty()){ ans += stack.peek(); } stack.push(i); } return ans; }