MATLAB中的find函数如何用C实现

MATLAB中的`find`函数用于返回向量或矩阵非零元素的索引。在C语言中，可以手动实现类似的功能，通常通过遍历数组并检查每个元素是否不为0来完成。以下是一个简单的C代码示例：

#include <stdio.h>
#include <stdbool.h>

int* find_nonzeros(int arr[], int n) {
    bool* is_zero = malloc(n * sizeof(bool));
    int* indices = NULL;
    int count = 0;

    for (int i = 0; i < n; i++) {
        is_zero[i] = (arr[i] == 0);
        if (!is_zero[i]) {
            count++;
            if (indices == NULL) {
                indices = malloc(count * sizeof(int));
                indices[count - 1] = i;
            } else {
                indices = realloc(indices, (count + 1) * sizeof(int));
                indices[count] = i;
            }
        }
    }

    return indices;
}

// 使用示例
int main() {
    int input[] = {0, 4, 0, 7, 0, 3};
    int size = sizeof(input) / sizeof(input[0]);

    int* result = find_nonzeros(input, size);
    for (int i = 0; result != NULL && i < size; i++, result++) {
        printf("Index %d: Non-zero element found at position %d\n", i, *result);
    }
    free(result);
    free(is_zero);

    return 0;
}

这个函数首先创建一个布尔数组记录非零元素位置，然后计算非零元素的数量，并动态分配内存存储索引。注意，在实际应用中，还需要处理可能出现的内存分配失败情况。