Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> C++

การรวม Vector สองอันด้วย STL ใน C++

การรวม Vector สองอัน (Join/Union Two Vectors) ด้วย STL ใน C++

ใช้ std::set_union จาก <algorithm> รวม vector สองอันที่เรียงแล้ว ได้ผลลัพธ์ที่เรียงและไม่ซ้ำ (union)

ตัวอย่าง

#include <bits/stdc++.h>
using namespace std;

int main() {
    vector<int> vector1 = { 1, 45, 54, 71, 76, 12 };
    vector<int> vector2 = { 1, 7, 5, 4, 6, 12 };
    
    // ต้องเรียงก่อนใช้ set_union
    sort(vector1.begin(), vector1.end());
    sort(vector2.begin(), vector2.end());
    
    cout << "First Vector: ";
    for (int x : vector1) cout << x << " ";
    cout << endl;
    
    cout << "Second Vector: ";
    for (int x : vector2) cout << x << " ";
    cout << endl;
    
    vector<int> v(vector1.size() + vector2.size());
    auto it = set_union(vector1.begin(), vector1.end(),
                        vector2.begin(), vector2.end(),
                        v.begin());
    
    cout << "\nAfter joining:\n";
    for (auto st = v.begin(); st != it; ++st)
        cout << *st << ", ";
    cout << '\n';
    return 0;
}

ผลลัพธ์

First Vector: 1 12 45 54 71 76
Second Vector: 1 4 5 6 7 12
After joining:
1, 4, 5, 6, 7, 12, 45, 54, 71, 76,

คำอธิบาย

  • set_union ต้องการ input ที่ เรียงแล้ว (sorted)
  • ผลลัพธ์เป็น sorted union (ไม่มีค่าซ้ำ)
  • คืนค่า iterator ชี้ท้ายของผลลัพธ์ (ใช้ resize หรือใช้ range constructor)
  • Complexity: O(n+m)

ฟังก์ชันเซตอื่นๆ ใน <algorithm>

  • set_intersection - ส่วนร่วม
  • set_difference - ความต่าง (A-B)
  • set_symmetric_difference - ความต่างสมมาตร
  • includes - ตรวจสอบ subset

หมายเหตุ

ถ้า input ไม่ได้เรียง ให้เรียงก่อน หรือใช้ std::unordered_set รวมแล้วค่อย copy กลับ vector