c++ matlab sort,Matlab的intersect函数的C++实现

Matlab的Intersect函数是求两个向量的交集,默认是对数据进行排序。

与setdiff函数类似,C++STL中也有set_intersection函数实现交集的功能,与setdiff转成C++类似,也需要几个步骤来得到最后一致的结果:

1、先对向量进行sort排序;

2、声明一个vector对象,大小为两个向量最长的那个值(保证向量的长度足够),使用set_intersection求交集,代码片段如下:

vector v;

vector::iterator it;

v.resize(ANumber>BNumber ? ANumber : BNumber);

it = set_intersection(arrayA[0], arrayA[0]+ANumber, arrayB[0], arrayB[0]+BNumber, v.begin());

3、set_intersection函数只是把两个向量的相同元素找出来并排列,不同的元素排列在最后一个相同元素后面,使得向量的长度并不变。如A[5]={1,1,2,3,4},B[3]={1,3,5},声明的vector是C,size是5,结果为C={1,1,3,2,4}

4、所以需要将vector截断,只保留相同元素的部分,利用第2步里的it,对vector进行resize:v.resize(it-v.begin());

5、对vector对象进行去重(unique)操作。与set_difference类似,转换成动态数组进行操作。

6、转换为自己的数据格式,得到和Matlab一样的结果。

整个代码如下,CIPPmatrix_I是自定义的二维数组:(应该有不必要的格式转换,还是对C++ STL不熟悉)

CIPPMatrix_I Method::InterSection(CIPPMatrix_I arrayA, CIPPMatrix_I arrayB) {     assert(arrayA.Height() == 1 && arrayA.Width() >= 1 && arrayB.Height() == 1 && arrayB.Width() >=1);     int ANumber = arrayA.Width();     int BNumber = arrayB.Width();     sort(arrayA[0],arrayA[0]+ANumber);     sort(arrayB[0], arrayB[0]+BNumber);     vector v;     vector::iterator it;     v.resize(ANumber>BNumber ? ANumber : BNumber);     it = set_intersection(arrayA[0], arrayA[0]+ANumber, arrayB[0], arrayB[0]+BNumber, v.begin());     v.resize(it-v.begin());          int* newarray = new int[v.size()];          int count = 0;     for(it=v.begin(); it!=v.end();++it)     {         newarray[count] = *it;         count++;     }     int* uniqueresult = unique(newarray, newarray+v.size());     vector la(newarray, uniqueresult);          CIPPMatrix_I result(1,la.size());     vector::iterator itunique;     count = 0;     for(itunique=la.begin(); itunique!=la.end();++itunique)     {         result[0][count] = *itunique;         count++;     }     delete[] newarray;     vector().swap(v);     vector().swap(la);     return result; }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值