题目
思路
①默认代码里需要自定义结构体、初始化、功能以及内存释放,故在结构体定义里可以同时定义好单调栈和一个用来记录当前结果的数组,以及栈顶指针;则在初始化里可以为存结果的数组申请内存,以及将栈顶指针初始化为 -1;
②在实现中:因为题目要找比当前小的股票价格(即以当前价格为基础,往前查找,如果找到比当前小于或等于的价格,则计入天数),故使用单调递减栈。栈顶指针永远指向当前找到的最大的元素(价格),当遇到比栈顶更大的元素时,则认为栈顶以及栈顶之前的元素都找到了答案(可以计算这些元素的价格跨度了),故将这些元素弹出栈,并且将结果数组更新。
代码实现
typedef struct {
int stack[10000];
int ans[10000];
int stackTop;
} StockSpanner;
StockSpanner* stockSpannerCreate() {
StockSpanner* obj = malloc(sizeof(StockSpanner));
obj->stackTop = -1;
return obj;
}
int stockSpannerNext(StockSpanner* obj, int price) {
int day = 1;
while (obj->stackTop >= 0 && price >= obj->stack[obj->stackTop]) {
day += obj->ans[obj->stackTop --];
}
obj->stack[++ obj->stackTop] = price;
obj->ans[obj->stackTop] = day;
return day;
}
void stockSpannerFree(StockSpanner* obj) {
free(obj);
}
/**
* Your StockSpanner struct will be instantiated and called as such:
* StockSpanner* obj = stockSpannerCreate();
* int param_1 = stockSpannerNext(obj, price);
* stockSpannerFree(obj);
*/