题目来自力扣,属于是在银行赞助的周赛推出了这一道每日一题,真刑啊!
今天也是作为解题回顾再次做这一题,这里附上题目:
你和一群强盗准备打劫银行。给你一个下标从 0 开始的整数数组 security ,其中 security[i] 是第 i 天执勤警卫的数量。日子从 0 开始编号。同时给你一个整数 time 。
如果第 i 天满足以下所有条件,我们称它为一个适合打劫银行的日子:
第 i 天前和后都分别至少有 time 天。
第 i 天前连续 time 天警卫数目都是非递增的。
第 i 天后连续 time 天警卫数目都是非递减的。
请你返回一个数组,包含 所有 适合打劫银行的日子(下标从 0 开始)。
返回的日子可以 任意 顺序排列。
第一次做这一道题的时候,一开始想的是从第time个元素开始遍历,一直到第numsSize-time个元素为止,这样就可以省下一定时间,但是这也增加了对边界判定的难度,需要深思熟虑访问越界的问题,最终在轰炸十多次堆区之后,我放弃了这一方法,选择了更加简单的标记算法,典型暴力加暴力的方法。
我开始尝试用两个数组,分别从左向右和从右向左维护非递增和非递减的天数:
int left[n],right[n];
memset(left,0,sizeof(left)); // 两个数组的初始化
memset(right,0,sizeof(right));
// 维护从左到右的非递增序列的长度
for(int i=1;i<n;i++)
{
if(security[i]<=security[i-1]) // 将符合条件的长度维护入数组中
{ // 如果不符合条件就从零开始重新计数
left[i]=left[i-1]+1;
}
}
// 维护从右到左的非递增序列的长度
for(int i=n-2;i>=0;i--)
{
if(security[i]<=security[i+1])
{
right[i]=right[i+1]+1;
}
}
通过这两次遍历,就可以找到当前元素向左和向右能够位置非递增和非递减的长度,只需要再遍历一次数组,找到符合条件的元素直接输出即可。
// 当两个序列的长度都大于等于time时, 即为符合条件的日子
int *ans=(int*)malloc(sizeof(int)*n);
int idx=0;
for(int i=0;i<n;i++) // 初始化循环的起点和终点,可以不用考虑起点和终点的优化
{
if(left[i]>=time&&right[i]>=time)
{
ans[idx++]=i; // 符合条件的直接计数存储
}
}
总结:方法不算最优解,毕竟使用了三次遍历,不会快到哪里去,但是这个整体的思路比较容易理解,做起来没有那么吃力。
最后附上整体代码和通过记录:
int* goodDaysToRobBank(int* security, int securitySize, int time)
{
int n=securitySize;
int left[n],right[n];
memset(left,0,sizeof(left)); // 两个数组的初始化
memset(right,0,sizeof(right));
// 维护从左到右的非递增序列的长度
for(int i=1;i<n;i++)
{
if(security[i]<=security[i-1]) // 将符合条件的长度维护入数组中
{ // 如果不符合条件就从零开始重新计数
left[i]=left[i-1]+1;
}
}
// 维护从右到左的非递增序列的长度
for(int i=n-2;i>=0;i--)
{
if(security[i]<=security[i+1])
{
right[i]=right[i+1]+1;
}
}
// 当两个序列的长度都大于等于time时, 即为符合条件的日子
int *ans=(int*)malloc(sizeof(int)*n);
int idx=0;
for(int i=0;i<n;i++) // 初始化循环的起点和终点,可以不用考虑起点和终点的优化
{
if(left[i]>=time&&right[i]>=time)
{
ans[idx++]=i; // 符合条件的直接计数存储
}
}
return ans;
}