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; }