刷题日记 1299. 将每个元素替换为右侧最大元素

73 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

1299. 将每个元素替换为右侧最大元素

给你一个数组 arr ,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -1替换。

完成所有替换操作后,请你返回这个数组。

 

示例 1:

输入:arr = [17,18,5,4,6,1]
输出:[18,6,6,6,1,-1]
解释:
- 下标 0 的元素 --> 右侧最大元素是下标 1 的元素 (18)
- 下标 1 的元素 --> 右侧最大元素是下标 4 的元素 (6)
- 下标 2 的元素 --> 右侧最大元素是下标 4 的元素 (6)
- 下标 3 的元素 --> 右侧最大元素是下标 4 的元素 (6)
- 下标 4 的元素 --> 右侧最大元素是下标 5 的元素 (1)
- 下标 5 的元素 --> 右侧没有其他元素,替换为 -1

示例 2:

输入:arr = [400]
输出:[-1]
解释:下标 0 的元素右侧没有其他元素。

提示:

1 <= arr.length <= 10^4 1 <= arr[i] <= 10^5

二、思路分析:

只要找右边的最大值,所以用逆序最快。 最大值这个点。设置一个容器max装最大值。 对于最后一个数来说,最大值就是他本身,对于倒数第二个,最大值是他右边的;对于倒数第三个,最大值是原本的max和倒数第二个数比较……以此类推。 没什么特别的,主要还是逆序遍历。

三、AC 代码:

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* replaceElements(int* arr, int arrSize, int* returnSize){

    int  max=arr[arrSize-1];
    int *res=(int *)malloc(sizeof(int)*arrSize);
    res[arrSize-1]=-1;

    for(int i=arrSize-2;i>=0;i--)
    {
        max=max<arr[i+1]?arr[i+1]:max;
        res[i]=max;
    }

    *returnSize=arrSize;
    return res;
}