花园里有 N 个花盆,每个花盆里都有一朵花。这 N 朵花会在 N 天内依次开放,每天有且仅有一朵花会开放并且会一直盛开下去。
给定一个数组 flowers 包含从 1 到 N 的数字,每个数字表示在那一天开放的花所在的花盆编号。
例如, flowers[i] = x 表示在第 i 天盛开的花在第 x 个花盆中,i 和 x 都在 1 到 N 的范围内。
给你一个整数 k,请你输出在哪一天恰好有两朵盛开的花,他们中间间隔了 k 朵花并且都没有开放。
如果不存在,输出 -1。
样例 1:
输入:
flowers: [1,3,2]
k: 1
输出: 2
解释: 在第二天,第一朵和第三朵花都盛开了。
样例 2:
输入:
flowers: [1,2,3]
k: 1
输出: -1
注释 :
给定的数组范围是 [1, 20000]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/k-empty-slots
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
简单分析:
记录每朵花开放的日期,比如flower:[1,3,2] 表示第一朵花开放的日期是第一天,第二朵花开放的日期是第三天,第三朵花开放的日期是第二天。
这样转化之后,题目的要求就变成,找到一个长度为K+2的窗口,使得中间的K个数全都大于两端的两个数(中间的花开放晚于两段的花)
时间复杂度大致为O(K*n)
class Solution {
public:
int kEmptySlots(vector<int>& bulbs, int K) {
int n=bulbs.size();
int* flower=new int[n];
for(int i=0;i<n;i++){
flower[bulbs[i]-1]=i;
}
int start=0;
int ans=INT_MAX;
while(start+K+1<n){
int someday=flower[start]>flower[start+K+1]?flower[start]:flower[start+K+1];
int flag=true;
for(int i=start+1;i<start+K+1;i++){
if(flower[i]<someday){
flag=false;break;
}
}
if(flag){
if(someday<ans) ans=someday;
}
start++;
}
if(ans!=INT_MAX) return ans+1;
else return -1;
}
};