Matlab的intersect函数的C++实现

来源:互联网 发布:农村淘宝面试和答案 编辑:程序博客网 时间:2024/06/03 19:42

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

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

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

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

vector<int> v;
    vector<int>::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<int> v;
    vector<int>::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<int> la(newarray, uniqueresult);
    
    CIPPMatrix_I result(1,la.size());
    vector<int>::iterator itunique;
    count = 0;
    for(itunique=la.begin(); itunique!=la.end();++itunique)
    {
        result[0][count] = *itunique;
        count++;
    }
    delete[] newarray;
    vector<int>().swap(v);
    vector<int>().swap(la);
    return result;
}

0 0
原创粉丝点击