C语言 leetcode 刷题篇适合打劫银行的日子(类动态规划题)

Frank.Ren

已于 2022-03-06 11:50:33 修改

阅读量478

点赞数 1

分类专栏： leetcode刷题系列 C_programs 后端文章标签： leetcode c语言算法

于 2022-03-06 11:47:36 首次发布

本文链接：https://blog.csdn.net/weixin_51666505/article/details/123307886

版权

leetcode刷题系列同时被 3 个专栏收录

90 篇文章 1 订阅

订阅专栏

C_programs

42 篇文章 0 订阅

订阅专栏

后端

8 篇文章 0 订阅

订阅专栏

这篇博客讨论了一个关于银行抢劫的问题，其中涉及到在给定警卫数量的时间序列中找到适合打劫的天数。算法工程师通过初始化left和right数组，并进行遍历来找出满足条件的连续时间段。该算法涉及数组操作、比较和动态规划思想，适用于解决区间查找和序列分析的问题。

摘要由CSDN通过智能技术生成

你和一群强盗准备打劫银行。给你一个下标从 0 开始的整数数组 security ，其中 security[i] 是第 i 天执勤警卫的数量。日子从 0 开始编号。同时给你一个整数 time 。

如果第 i 天满足以下所有条件，我们称它为一个适合打劫银行的日子：

第 i 天前和后都分别至少有 time 天。
第 i 天前连续 time 天警卫数目都是非递增的。
第 i 天后连续 time 天警卫数目都是非递减的。
更正式的，第 i 天是一个合适打劫银行的日子当且仅当：security[i - time] >= security[i - time + 1] >= ... >= security[i] <= ... <= security[i + time - 1] <= security[i + time].

请你返回一个数组，包含所有适合打劫银行的日子（下标从 0 开始）。返回的日子可以任意顺序排列。

新建left,right数组来进行最后的判断

在初始化这两个数组时,关键的两步就是:

memset(left, 0, sizeof(int) * securitySize);
memset(right, 0, sizeof(int) * securitySize);

用来将left和right各位都设为零,这样在判断时,if为否,也不需要再将这一位设置成0,因为原本没有填充时,本来就是零,相当于省去了重置这一步.

if (security[i] <= security[i - 1]) {
    left[i] = left[i - 1] + 1;
}
if (security[securitySize - i - 1] <= security[securitySize - i]) {
    right[securitySize - i - 1] = right[securitySize - i] + 1;
}

完整代码如下:

int* goodDaysToRobBank(int* security, int securitySize, int time, int* returnSize) {
    int * left = (int *)malloc(sizeof(int) * securitySize);
    int * right = (int *)malloc(sizeof(int) * securitySize);
    memset(left, 0, sizeof(int) * securitySize);
    memset(right, 0, sizeof(int) * securitySize);
    for (int i = 1; i < securitySize; i++) {
        if (security[i] <= security[i - 1]) {
            left[i] = left[i - 1] + 1;
        }
        if (security[securitySize - i - 1] <= security[securitySize - i]) {
            right[securitySize - i - 1] = right[securitySize - i] + 1;
        }
    }

    int * ans = (int *)malloc(sizeof(int) * securitySize);
    *returnSize = 0;
    for (int i = time; i < securitySize - time; i++) {
        if (left[i] >= time && right[i] >= time) {
            ans[(*returnSize)++] = i;
        }
    }
    free(left);
    free(right);

    return ans;
}