가능한 한 적은 수의 루프에 데이터를 동적으로 추가

Question

나는 행렬 A=(n,m)을 가지고 있으며 열을 하나씩 스캔해야하며 열 요소의 합계가 임계 값보다 큰 경우 열을 새 행렬에 넣어야합니다 B=(n,?). 분명히

B=[1 3 
    3 1 
    4 8] 

, 내가 따라서 첫 번째 루프에서, 나는 그 값이 경우 열의 요소의 합을 확인 B에 얼마나 많은 열을 모르는 :

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

threshold=6 

그리고 그 결과로

가 임계 값보다 큰 경우 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

링크 된 목록을 사용할 수 있습니다. 따라서 요소 (이 경우 임계 값을 충족하는 모든 열의 인덱스)를 추가 할 때마다주기가 끝나면 할당해야 할 메모리 양을 알 수 있습니다 열을 간단한 2 차원 배열에 넣습니다.

최소한이 방법으로 재 할당해야합니다.

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