考察知识点:数组、单调栈
题目分析:
方法一:双指针
遍历一遍数组即可。指针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; } };