LeetCode 1124. 表现良好的最长时间段题解 C/C++

最新推荐文章于 2022-09-28 15:47:23 发布

【止心】

最新推荐文章于 2022-09-28 15:47:23 发布

阅读量219

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：单调栈前缀和 c++

本文链接：https://blog.youkuaiyun.com/Jason__Jie/article/details/117173076

LeetCode 专栏收录该内容

91 篇文章

订阅专栏

该博客介绍了如何利用前缀和和单调栈数据结构解决类似于LeetCode 962题的问题。通过将输入小时数转换为1（超过8小时）或0，并构建前缀和数组，然后利用单调栈找到最大宽度。算法首先计算小时数的前缀和，接着创建一个严格单调递减的栈，最后从前向后遍历找到最大宽度。这种方法展示了在算法问题中如何巧妙地应用数据结构来优化解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

类比LeetCode 962. 最大宽度坡

计大于8小时为1，否则为0，生成前缀和数组，该前缀和数组就是962题中给定的数组，计算最大宽度，对前缀和数组做单调栈（严格单调递减）。

//前缀和+单调栈
class Solution {
public:
    int longestWPI(vector<int>& hours) {
    	int len = hours.size();
    	vector<int> prefix_sum(len+1,0);//前缀和，第一个是0

    	for(int i = 1;i<=len;++i) {
    		prefix_sum[i] = (hours[i-1] > 8 ? 1 : -1) + prefix_sum[i-1];
    	}

    	stack<int> dec_stk;//严格单调递减栈
    	for(int j = 0;j<=len;++j) {
    		if(dec_stk.empty() || prefix_sum[j] < prefix_sum[dec_stk.top()]) {
    			dec_stk.push(j);
    		}
    	}

    	//从后往前遍历 确定最长时间段
    	int ans = 0;
    	for(int k = len;k>ans;--k) {
    		while(!dec_stk.empty() && prefix_sum[k] > prefix_sum[dec_stk.top()]) {
    			ans = max(ans,k - dec_stk.top());
    			dec_stk.pop();
    		}
    	}
    	return ans;
		
    }
};