C++ 实现 MATLAB ismember 函数原理与代码解析

C++ 实现 MATLAB ismember 函数原理与代码解析

环境配置

要在 C++ 环境中复现 MATLAB 的 ismember 功能，首先需要准备好基础库文件。确保项目能够链接到 Matlab2c 相关的动态库和头文件。

1. 获取依赖：确认已下载 Matlab2c.dll、Matlab2c.h 及 Matlab2c.lib。
2. 路径设置：将 .dll 文件放置于可执行文件同级目录，.h 和 .lib 加入项目的包含路径与链接库路径。
3. 引入头文件：在源文件中添加必要的引用。

#include "Matlab2c.h"
#pragma comment(lib, "Matlab2c.lib")
using namespace Matlab2c;

函数逻辑分析

ismember 的核心作用是检测一个集合中的元素是否存在于另一个集合中。在 MATLAB 中，它返回两个向量：一个是布尔标记（存在为 1，不存在为 0），另一个是位置索引（存在则返回下标，不存在则为 0）。

在我们的 C++ 实现中，为了更贴近底层逻辑，我们将返回值设计为：第一个向量记录存在性（1/0），第二个向量记录位置索引（存在为下标，不存在为 -1）。输入参数为两个行向量（一维矩阵），输出为指向结果向量的指针数组。

C++ 核心实现

这里采用双重循环遍历比对的方式。外层遍历集合 A 的元素，内层在集合 B 中查找匹配项。虽然对于大数据集这种 O(N*M) 的复杂度不是最优解，但作为算法原型非常直观。

Matrix* Matlab2c::ismember(Matrix& a, Matrix& b) {
    bool exist = false;
    Matrix *p;
    // p1 记录 a 中的元素是否存在于 b 中，存在为 1，不存在为 0
    Matrix p1(1, a.column);
    // p2 记录 a 中的元素在 b 中的位置，不存在则为 -1
    Matrix p2(1, a.column);
    
    p = new Matrix[2];
    
    for (int j = 0; j < a.column; j++) {
        int n = ;
        exist = ;
        
         (n = ; n < b.column; n++) {
             (a.data[j] == b.data[n]) {
                exist = ;
                ;
            }
        }
        
         (exist) {
            pdata[j] = ;
            pdata[j] = n;
        }  {
            pdata[j] = ;
            pdata[j] = ;
        }
    }
    
    *p = p1;
    *(p + ) = p2;
     p;
}