在 C++ 中從向量中刪除重複項

Jinku Hu 2023年10月12日
  1. 使用 std::uniquestd::vector::erase 函式從 C++ 向量中刪除重複項
  2. 使用 std::set 容器從 C++ 向量中刪除重複項
在 C++ 中從向量中刪除重複項

本文將介紹如何在 C++ 中去除向量中的重複項。

使用 std::uniquestd::vector::erase 函式從 C++ 向量中刪除重複項

std::unique 函式是 STL 演算法的一部分,它本質上實現了對排序範圍的重複元素刪除操作。每次刪除後元素都會移動,並且該函式返回新形成的範圍的最後迭代器。我們在以下示例程式碼中使用的 std::unique 過載採用兩個迭代器引數來表示範圍的開始和結束。在這種情況下,我們還使用 std::sort 演算法對給定的 vector 進行排序,然後再使用 std::unique 對其進行處理。最後,呼叫 erase 成員函式來修改原始陣列大小以適應新形成的向量。

#include <algorithm>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <vector>

using std::copy;
using std::cout;
using std::endl;
using std::left;
using std::setw;
using std::vector;

int main() {
  vector<int> int_vec = {10,  23,  10,  324, 10, 10, 424,
                         649, 110, 110, 129, 40, 424};

  cout << left << setw(10) << "vec: ";
  copy(int_vec.begin(), int_vec.end(), std::ostream_iterator<int>(cout, "; "));
  cout << endl;

  std::sort(int_vec.begin(), int_vec.end());
  auto last = std::unique(int_vec.begin(), int_vec.end());
  int_vec.erase(last, int_vec.end());

  cout << left << setw(10) << "vec: ";
  copy(int_vec.begin(), int_vec.end(), std::ostream_iterator<int>(cout, "; "));
  cout << endl;

  return EXIT_SUCCESS;
}

輸出:

vec:      10; 23; 10; 324; 10; 10; 424; 649; 110; 110; 129; 40; 424;
vec:      10; 23; 40; 110; 129; 324; 424; 649;

實現上述解決方案的另一種方法是使用 resize 函式而不是 erase 呼叫。resize 函式修改向量的元素計數。因此,我們可以使用 distance 呼叫傳遞新形成的向量元素的計算計數,而 resize 將縮小向量。

#include <algorithm>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <vector>

using std::copy;
using std::cout;
using std::endl;
using std::left;
using std::setw;
using std::vector;

int main() {
  vector<int> int_vec = {10,  23,  10,  324, 10, 10, 424,
                         649, 110, 110, 129, 40, 424};

  cout << left << setw(10) << "vec: ";
  copy(int_vec.begin(), int_vec.end(), std::ostream_iterator<int>(cout, "; "));
  cout << endl;

  std::sort(int_vec.begin(), int_vec.end());
  auto last = std::unique(int_vec.begin(), int_vec.end());
  int_vec.resize(std::distance(int_vec.begin(), last));

  cout << left << setw(10) << "vec: ";
  copy(int_vec.begin(), int_vec.end(), std::ostream_iterator<int>(cout, "; "));
  cout << endl;

  return EXIT_SUCCESS;
}

輸出:

vec:      10; 23; 10; 324; 10; 10; 424; 649; 110; 110; 129; 40; 424;
vec:      10; 23; 40; 110; 129; 324; 424; 649;

使用 std::set 容器從 C++ 向量中刪除重複項

或者,std::set 容器可用於從向量中刪除重複元素。請注意,std::set 在內部儲存給定型別的唯一物件,因此我們需要從向量元素構造一個。一旦 set 用需要排序的向量元素初始化,我們可以從原始 vector 物件呼叫 assign 函式來儲存來自 set 物件的唯一元素。

#include <algorithm>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <set>
#include <vector>

using std::copy;
using std::cout;
using std::endl;
using std::left;
using std::set;
using std::setw;
using std::vector;

int main() {
  vector<int> int_vec = {10,  23,  10,  324, 10, 10, 424,
                         649, 110, 110, 129, 40, 424};

  cout << left << setw(10) << "vec: ";
  copy(int_vec.begin(), int_vec.end(), std::ostream_iterator<int>(cout, "; "));
  cout << endl;

  set<int> int_set(int_vec.begin(), int_vec.end());
  int_vec.assign(int_set.begin(), int_set.end());

  cout << left << setw(10) << "vec: ";
  copy(int_vec.begin(), int_vec.end(), std::ostream_iterator<int>(cout, "; "));
  cout << endl;

  return EXIT_SUCCESS;
}

輸出:

vec:      10; 23; 10; 324; 10; 10; 424; 649; 110; 110; 129; 40; 424;
vec:      10; 23; 40; 110; 129; 324; 424; 649;
作者: Jinku Hu
Jinku Hu avatar Jinku Hu avatar

DelftStack.com 創辦人。Jinku 在機器人和汽車行業工作了8多年。他在自動測試、遠端測試及從耐久性測試中創建報告時磨練了自己的程式設計技能。他擁有電氣/ 電子工程背景,但他也擴展了自己的興趣到嵌入式電子、嵌入式程式設計以及前端和後端程式設計。

LinkedIn Facebook

相關文章 - C++ Vector