动态数据附加在尽可能少的回路尽可能

Question

我具有矩阵A=(n,m)和我需要逐个并且如果列的元素的总和为大于阈值来扫描它的列，把柱到一个新的矩阵B=(n,?)。

A=[1 2 3 
    3 1 1 
    4 2 8] 

threshold=6 

而作为结果：

B=[1 3 
    3 1 
    4 8] 

很显然，我不知道有多少列B，因此在第一循环中，我检查元素之和在一列，那么如果该值大于阈值I，重复将B中的列附加realloc的循环。

okcol=0; 
double *B = malloc(n*sizeof(double)); 
for (col=0;col<m;col++){ 
    sum=0; 
    for (row=0;row<n;row++){ 
    sum+=A[row+col*n]; 
    if(sum>threshold){ 
     B = realloc(B, (okcol+1)*n*sizeof(double)); 
     for (row2=0;row2<n;row2++){ 
     B [okcol*n+row2] = A[row2+col*n]; 
     } 
     okcol++; 
    } 
    } 
}   

有没有一种方法可以自动补齐元素B和“删除”他们如果总和小于阈值？换句话说，我想避免最后一个循环。

Answer 1

你可以使用一个链表，所以每次你只需要添加一个元素（在这种情况下，每一个达到阈值列的索引），并在周期结束时，你知道你有多少内存分配给将列放入一个简单的二维数组中。

这样你至少会有很多重新分配。

Answer 2

double *B = (double*)malloc(n*m*sizeof(double)); 
int c,r,okcol=0; 
for(c=0;c<m;++c){ 
    for(sum=0, r=0;r<n;++r){ 
     sum+=(B[r+okcol*m]=A[r+c*m]); 
    } 
    if(sum > threshold) 
     ++okcol; 
} 
B=(double*)realloc(B, okcol*m*sizeof(double));