leetcode：从排序数组中删除重复项（scala）

题目

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

注意项

1. 排序数组
2. 复杂度为O(1)
3. 不额外使用空间

 scala

object Solution {
  def removeDuplicates(nums: Array[Int]): Int = {
    var j = 0
    if (nums.length == 0) {
      0
    } else {
      for (i <- 1 to nums.length - 1) {
        if (nums(j) != nums(i)) {
          j = j + 1
          nums(j) = nums(i)
        }
      }
      j + 1
    }
  }
}

Java

class Solution {
    public int removeDuplicates(int[] nums) {
        int j = 0;
        if (nums.length == 0) {
            return 0;
        } else {
            for (int i = 1; i < nums.length; i++) {
                if (nums[j] != nums[i]) {
                    j++;
                    nums[j] = nums[i];
                }
            }
            return ++j;
        }
    }
}