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

你和一群强盗准备打劫银行。给你一个下标从 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;
}