题解 | #牛的生长情况#

牛的生长情况

https://www.nowcoder.com/practice/5f67258999bd4e61a361f4d3017a3fd4

考察知识点:数组、单调栈

题目分析:

方法一:双指针

遍历一遍数组即可。指针i指向第i天,用指针j向后查找看那一天能大于第i天的体重,返回j与i的差值即可。没有找到则返回-1。

方法二:单调栈

单调栈适合解决这种找与某个数最近的更大的数或更小的数的问题。

如果我们想要找到右边序列中更大或更小的数,那么就是说我需要先知道右边的数,即我需要从右向左遍历。

如上图所示,第一个遍历到的数是13,他右边没有数,所以它对应的grouth就是-1。而第二个遍历到的数是76,右边的数是13比他小,所以它对应的grouth是-1。第三个数是72,右边的数是76和13,此时发现左边数的grouth不可能取到13了,因为76比13要大,76总是先被取到。

也就是说,如果将所有遍历到的数放入栈中,如果指针i指向的数比栈顶大,那么就不断将栈顶删除,直到找到栈中的数比指针i要大即可:

上图中,71能很快找到右边比他大的最近的数是72,75右边比它大的数是76。

注意栈中应该存放每个数的坐标,方便计算天数之差。

所用编程语言:C++

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param weights int整型vector 
     * @return int整型vector
     */
    vector<int> weightGrowth(vector<int>& weights) {
        // write code here
        int size = weights.size();
        vector<int> res;
        for (int i = 0; i < size; i++) {
            for (int j = i + 1; j < size; j++) {
                if (weights[j] > weights[i]) {
                    res.push_back(j - i);
                    break;
                }
                if (j == size - 1 && weights[j] <= weights[i]) {
                    res.push_back(-1);
                }
            }
        }
        res.push_back(-1);
        return res;
    }
};
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param weights int整型vector 
     * @return int整型vector
     */
    vector<int> weightGrowth(vector<int>& weights) {
        // write code here
        int size = weights.size();
        vector<int> res(size);
        stack<int> stk;
        for (int i = size - 1; i >= 0; i--) {
            while (!stk.empty() && weights[stk.top()] <= weights[i]) stk.pop();
            if (!stk.empty()) res[i] = stk.top() - i;
            else res[i] = -1;
            stk.push(i);
        }
        return res;
    }
};

全部评论
你的题解好认真
点赞 回复 分享
发布于 2023-08-22 19:23 北京

相关推荐

Beeee0927:正确的建议
点赞 评论 收藏
分享
04-30 15:51
已编辑
上海交通大学 机械工程师
点赞 评论 收藏
分享
评论
4
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务