【每日一题Day227】LC2460对数组执行操作 | 模拟+双指针

对数组执行操作【LC2460】

给你一个下标从 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] 。

返回结果数组。

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

• 思路

  简单模拟题，执行完所有操作后，使用双指针将非0的元素移动数组前端，同时将末尾元素赋值为0

• 实现

  class Solution {
        
        
      public int[] applyOperations(int[] nums) {
        
        
          int n = nums.length;
          for (int i = 0; i < n - 1; i++){
        
        
              if (nums[i] == nums[i + 1]){
        
        
                  nums[i] *= 2;
                  nums[i + 1] = 0;
              }
  
          }
          int i = 0, j = 0;
          while(j < n){
        
        
              nums[i] = nums[j];
              if (i != j){
        
        
                  nums[j] = 0;
              }
              if (nums[i] != 0){
        
        // 不为0移动指针；为0不移动
                  i++;
              }
              j++;                  
          }
          return nums;
      }
  }
  

  • 复杂度
    • 时间复杂度：$\mathcal{O}(n)$
    • 空间复杂度：$\mathcal{O}(1)$