MATLAB的union函数是将两个向量合并,同时去除重复元素,并按顺序排列。
而C++ STL中的set_union仅仅是将向量合并,即没有去重,也没有按序排列。
所以需要在set_union的基础上进行处理:
以float类型的向量为例:
float* A; float* B;//两个向量
int numA, numB;//两个向量的长度
//先排序
std::sort(A,A+numA);
std::sort(B,B+numB);
vector v;
v.resize(numA+numB);//先按最大长度声明
vector::iterator it;
it = set_union(A, A+numA, B, B+numB, v.begin()); //结果存在V里,但包含有重复元素
v.resize(it - v.begin());//去除多余元素,v中多余的位置存放0.0
//转换为float数组且统计数量
float* newarray = new float[v.size()];
int count = 0;
for(it=v.begin(); it!=v.end();++it)
{
newarray[count] = *it;
count++;
}
//对结果进行unique操作,清除相邻的重复值
float* index = unique(newarray, newarray + count);
//得到最终有效个数
int countnum = index - newarray;
float* C = new float[countnum]; //存放最后的结果
memcpy(C,newarray, countnum*sizeof(float));