删除已排序数组中重复的数字

删除已排序数组中重复的数字

问题描述

给定排序数组，删除重复项，使每个元素只出现一次并返回新长度。
例如：已知数组输入数组为A [1, 1, 2]， 输出为A[1, 2]

程序

#include <iostream>
#include <vector>

int removeDuplicatesElem(std::vector<int>& nums);
void print(std::vector<int>& nums, const std::string& note="");

int main()
{
  std::cout << "please input a array: " << std::endl;
  int elem;
  std::vector<int> nums;
  while (std::cin >> elem)
  {
    nums.push_back(elem);
  }
  print(nums, "output origin array:" );

  std::cout << "\nresult: " << std::endl;

  int n = removeDuplicatesElem(nums);
  for (int i  = 0; i < n; ++i)
    std::cout << nums[i] << ", ";
  std::cout << std::endl;

  return 0;
}

void print(std::vector<int>& nums, const std::string& note)
{
  std::cout << std::endl << note << std::endl;
  for (int i = 0; i < nums.size(); ++i)
  {
    std::cout << nums[i] << " ";
  }
}

int removeDuplicatesElem(std::vector<int>& nums)
{
  if (nums.empty())
    return 0;

  int index = 0;
  for (int i = 1; i < nums.size(); ++i)
  {
    if (nums[index] != nums[i])
    {
      nums[++index]  = nums[i];
    }
  }
  return index + 1;
}