牛客网NC80-20.7.21-栈？

枪打出头鸟算法优化

最新推荐文章于 2023-04-23 16:02:04 发布

原创最新推荐文章于 2023-04-23 16:02:04 发布 · 128 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈

本文探讨了在大规模数据集上解决枪打出头鸟问题的高效算法。通过使用栈结构，避免了传统暴力搜索的高时间复杂度，实现了线性时间复杂度的解决方案。该算法针对每个个体，快速找到其前方最近且更高的个体，适用于处理高达1e7规模的数据。

题意：枪打出头鸟，n个人，数组a, ai-1代表身高，求每个人前面离他最近地、身高比他高地的下标之和。
输入：n,a(1<=n<=1e7,1<=a[i]<=1e7)
输出：如题意的下标之和。
分析：每个人对于他后面的人来说会屏蔽前面比他矮的人，相对于暴力查找，可以通过栈来省略这部分的寻找。
思路：暴力O(n²）:双层循环，对于每一个人，从自己前一个开始往前遍历，遇到身高比自己高的记录跳出。但这题能过。栈O(n)：定义一个栈，遇到一个人，判断栈顶是否比他高，不是出栈直到栈空或者比他高的人出现,如果栈不空即是有比他高的，记录，将他入栈。
代码：

import java.util.*;


public class Solution {
    /**
     * 
     * @param n int整型 n个人
     * @param a int整型一维数组 ai代表第i个人的高度
     * @return long长整型
     */
    
    public long solve (int n, int[] a) {
        // write code here
        
        Stack<Integer> s=new Stack();
        long ans=0L;
        for(int i=0;i<n;i++){
            while(!s.empty()&&a[s.peek()]<=a[i])s.pop();
            if(!s.empty())ans+=s.peek()+1;
            s.add(i);
        }
        return ans;
    }
}