MATLAB中union函数的C++实现

来源:互联网 发布:javascript 取2位小数 编辑:程序博客网 时间:2024/06/07 13:17

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<float>  v;

v.resize(numA+numB);//先按最大长度声明

vector<float>::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));


0 0
原创粉丝点击