Question

我正在尝试找到2个排序数组（带有重复数据）的联合但我觉得我没有想出最优雅的代码（我的工作顺便说一下，我觉得我可以减少一些代码行）。假设我有2个向量a = {1,3,3,4,4,4,5,7}和b = {1,3,3,3,5,5,5,6,8,9}和我想将他们的联合存储在一个名为unionVector的向量中（将是1,3,4,5,6,7,8,9）

这是我的代码：

#include <iostream>
#include <vector>
using namespace std;

// Prints the contents of a vector
void printVector(vector<int> a){
  if(a.size() == 0)
    return;
  else{
    for(int i = 0; i < a.size(); i++)
      cout << a[i] << '\t';
  }
  cout << endl;
}

// Print the union of 2 sorted arrays with duplicates
void printUnion(int *a, int aSize, int *b, int bSize){
  if(aSize == 0 && bSize == 0)
    return;
  else{

    vector<int> unionVector;

    int i = 0;
    int j = 0;
    int last = 0;

    // insert the smaller of first element regardless
    if(a[i] < b[j]){
      unionVector.push_back(a[i]);
      i++;
    }
    else if (b[j] < a[i]){
      unionVector.push_back(b[j]);
      j++;
    }
    else{// both are equal numbers 
      unionVector.push_back(a[i]);
      i++;
      j++;
    }

    // now traverse both the loops one increment at a time
    while(i < aSize && j < bSize){
      last = unionVector[unionVector.size() - 1];

      if(a[i] < b[j]){
        if(last != a[i])
          unionVector.push_back(a[i]);
        i++; // increment i in either case
      }
      else if(b[j] < a[i]){
        if(last != b[j])
          unionVector.push_back(b[j]);
        j++;
      }
      else{
        // both of the numbers are equal
        if(last != a[i])
          unionVector.push_back(a[i]);
        i++;
        j++;
      }
    }

    // lets say if 1 array wasn't complete
    while(i < aSize){
      last = unionVector[unionVector.size() - 1];

      if(last != a[i])
        unionVector.push_back(a[i]);
      i++;
    }

    while(j < bSize){
      last = unionVector[unionVector.size() - 1];

      if(last != b[i])
        unionVector.push_back(b[j]);
      j++;
    }

    printVector(unionVector);
  }
}

int main(){
  int a[] = {1,3,3,4,4,4,5,7};
  int b[] = {1,3,3,3,5,5,5,6,7,7,8,9};

  printUnion(a,8,b,12);

  return 0;
}

因为可能存在重复，所以我检查要插入的元素以及插入unionVector中的最后一个元素。我需要确保当unionVector为空时我不会尝试获取'last'元素，这就是为什么我要在unionVector中插入1个元素。我真的很感激，如果有人可以提出一种方法，我可以做检查而不需要先插入1个元素（我正在考虑有一个标志变量来检查unionVector是否为空，但我觉得这样太乱了）

编辑1：

这不是作业问题。这是我在面试时练习的内容

编辑2：

我也不能使用任何内置函数

编辑3：

如果这是一个C ++职位，有些人会感到困惑。您可以使用任何您想要的语言

Answer 1

如果两个数组都已排序，那么只需跳过一个迭代器或另一个或两者，如果匹配的话。

类似于：

void printUnion(int* a, int aSize, int* b, int bSize)
{
    int *aEnd = a + aSize, *bEnd = b + bSize;
    std::vector<int> unionVec;

    for (; a != aEnd; ) {
        if (b == bEnd) {
            // copy all of a
            while (a != aEnd) {
                unionVec.push_back(*a);
                a = std::upper_bound(a + 1, aEnd, *a);
            }
            break;
        }

        if (*b < *a) {
            unionVec.push_back(*b);
            b = std::upper_bound(b + 1, bEnd, *b);
        } 
        else {
            unionVec.push_back(*a);
            if (*b == *a) {
                b = std::upper_bound(b + 1, bEnd, *b);
            }
            a = std::upper_bound(a + 1, aEnd, *a);
        }
    }

    // copy all of b
    while (b != bEnd) {
        unionVec.push_back(*b);
        b = std::upper_bound(b + 1, bEnd, *b);
    }

    printVector(unionVec);
}

如果您无法直接使用upper_bound，请自行实施该功能。从this reference：

复制实施

template<class ForwardIt, class T>
int* upper_bound(int* first, int* last, const int value)
{
    int* it;
    int count = last - first;
    int step;

    while (count > 0) {
        it = first; 
        step = count / 2; 
        it += step;
        if (value >= *it) {
            first = ++it;
            count -= step + 1;
        }
        else {
            count = step;
        }
    }

    return first;
}

或非二进制搜索版本：

int* upper_bound(int* first, int* last, const int value) {
    for (; first < last && *first == value; ++first) {
        ;
    }

    return first;
}

现在这显然非常冗长，这就是为什么标准实际上直接为您提供了算法set_union：

void printUnion(int* a, int aSize, int* b, int bSize)
{
    std::vector<int> unionVec;

    // get the union
    std::set_union(a, a + aSize, b, b + bSize, std::back_inserter(unionVec));

    // remove the dupes
    unionVec.erase(std::unique(unionVec.begin(), unionVec.end()), unionVec.end());

    printVector(unionVec);
}

Answer 2

这是一种方法。优雅可能会有所不同！

void printUnion(int* a, int aSize, int* b, int bSize)
{
    std::multiset<int> x;
    x.insert(a, a + aSize);
    x.insert(b, b + bSize);

    for (auto y : x )
        cout << y << ",";
    cout << endl;
}

NB。考虑让printUnion采用迭代器对。使用std::set忽略重复项，或std::multiset保留重复项。

找到2个排序数组的联合（带有重复数据）

2 个答案: