题目描述
给你一份工作时间表 hours,上面记录着某一位员工每天的工作小时数。我们认为当员工一天中的工作小时数大于 8 小时的时候,那么这一天就是「劳累的一天」。所谓「表现良好的时间段」,意味在这段时间内,「劳累的天数」是严格 大于「不劳累的天数」。请你返回「表现良好时间段」的最大长度。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/longest-well-performing-interval
输入
输入:hours = [9,9,6,0,6,6,9]
输出:3
解释:最长的表现良好时间段是 [9,9,6]。
输出
输入:hours = [6,6,6]
输出:0
提示
1 <= hours.length <= 104
0 <= hours[i] <= 16
暴力解法
思路
- 整体思路:将大于8的数字置为1,小于8的数字置为-1;求n个数的前缀和;题目要求返回表现良好时间段的最大长度,即求最长的一段中,得分 1 的个数大于得分 −1 的个数,也就是求 sum 数组中最长的一段子数组,其和大于 0,那么也就是找出前缀和数组 sum 中两个索引 i 和 j,使 j - i 最大,且保证 presum[j] - presum[i] 大于 0。
- 求n个数的前缀和sum,sum.length=n+1。这样做有两个目的,一个是防止i-1的时候地址越界,二是计算时间长度的时候,若是从第一天到第三天,一共有3天,但是3-1=2,我们还要加上1,因此,我们置sum[0]=0,这样相当于把时间长度+1,不用再考虑+1的事情。
- 计算sum[j]-sum[i]>0,如果满足,说明[i,j]这个时间段符合表现良好时间段(这里j是倒序比较,因为发生在后面的几率更大)。
- 在所有时间段[i,j]中求最长的。
前缀和请参考https://blog.csdn.net/z164470/article/details/126338187?spm=1001.2014.3001.5501
#include <iostream>
#include <vector>
using namespace std;
class Solution {
//求两个数中的最大值
int max(int a,int b){
if(a>b) return a;
else return b;
}
public:
//求最长的良好时间段
int longestWPI(vector<int>& hours) {
int n = hours.size(), nums = 0;
int* sum = new int[n+1];
sum[0] = 0;
for(int i=1;i<=n;i++){
sum[i] = sum[i-1]+(hours[i-1] > 8 ? 1: -1);
}
for (int i = 0; i <= n; i++) {
for (int j = n; j > i; j--) { // 反着查找,找到了可以continue跳出
if (sum[j] > sum[i]) {
nums = max(nums, j - i);
continue;
}
}
}
return nums;
}
};
int main()
{
vector<int> v = {6,9,6,6,6,9,9,9,9,9,6,6,6,6,9};
Solution s;
cout << s.longestWPI(v);
}
时间复杂度O(n*n)
空间复杂度O(n)