链接:https://leetcode-cn.com/problems/longest-well-performing-interval/
将大于8的元素看做1,小于等于8的看做-1。遍历数组计算前缀和,若当前前缀和大于0,则最长时间段必为当前全部时间;若小于0,则为最前一个 当前前缀和-1 到现在。
java代码:
class Solution {
public int longestWPI(int[] hours) {
Map<Integer,Integer>map = new HashMap();
int max = 0; //当前最大值
int sum = 0; //前缀和
map.put(0,-1);
for(int i = 0;i<hours.length;i++)
{
sum+= (hours[i] >8)? 1:-1;
if(sum>0)
max = i+1;
else
{
if(map.containsKey(sum-1))
max = Math.max(max,i-map.get(sum-1));
if(!map.containsKey(sum))
map.put(sum,i);
}
}
return max;
}
}