力扣2460


2460. 对数组执行操作

给你一个下标从 0 开始的数组 nums ,数组大小为 n ,且由 非负 整数组成。

你需要对数组执行 n - 1 步操作,其中第 i 步操作(从 0 开始计数)要求对 nums 中第 i 个元素执行下述指令:

如果 nums[i] == nums[i + 1] ,则 nums[i] 的值变成原来的 2 倍,nums[i + 1] 的值变成 0 。否则,跳过这步操作。
在执行完 全部 操作后,将所有 0 移动 到数组的 末尾 。

例如,数组 [1,0,2,0,0,1] 将所有 0 移动到末尾后变为 [1,2,1,0,0,0] 。
返回结果数组。

注意 操作应当 依次有序 执行,而不是一次性全部执行。

示例 1:

输入:nums = [1,2,2,1,1,0]
输出:[1,4,2,0,0,0]
解释:执行以下操作:

  • i = 0: nums[0] 和 nums[1] 不相等,跳过这步操作。

  • i = 1: nums[1] 和 nums[2] 相等,nums[1] 的值变成原来的 2 倍,nums[2] 的值变成 0 。数组变成 [1,4,0,1,1,0] 。

  • i = 2: nums[2] 和 nums[3] 不相等,所以跳过这步操作。

  • i = 3: nums[3] 和 nums[4] 相等,nums[3] 的值变成原来的 2 倍,nums[4] 的值变成 0 。数组变成 [1,4,0,2,0,0] 。

  • i = 4: nums[4] 和 nums[5] 相等,nums[4] 的值变成原来的 2 倍,nums[5] 的值变成 0 。数组变成 [1,4,0,2,0,0] 。
    执行完所有操作后,将 0 全部移动到数组末尾,得到结果数组 [1,4,2,0,0,0] 。

  • 双指针排序

  • 如果为零j指针不动,i继续循环,如果i不为0了,就交换i和j的值。

class Solution:
    def mysort(self, nums):
        temp = []
        lens_nums = len(nums)
        for i in nums:
            if i != 0:
                temp.append(i)
        zero_num = lens_nums - len(temp)
        for i in range(zero_num):
            temp.append(0)
        return temp

    def applyOperations(self, nums: List[int]) -> List[int]:
        # print(nums)
        # print(len(nums))
        for i in range(len(nums)):
            if i+1 == len(nums):
                nums = self.mysort(nums)
                break
            if nums[i] == nums[i+1]:
                nums[i] = nums[i] * 2
                nums[i+1] = 0
        return nums

# nums = [1,2,2,1,1,0]
# s = Solution()
# s.applyOperations(nums)
# # print(nums)

文章作者: 小冷同学
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 小冷同学 !
用户交流区

温馨提示: 遵纪守法, 友善评论!