Найти столбцы матрицы, которые содержат элементы равные 1

int counter = 0; 
for (int i = 0; i < n; i++) 
{ 
for (int j = 0; j < m; j++) 
{ 
if (a[i][j] == 1) 
counter++; 
} 
} 
int* res = new int[counter]; 
int c = 0; 
for (int i = 0; i < n; i++) 
{ 
for (int j = 0; j < m; j++) 
{ 
if (a[i][j] == 1) 
{ 
res[c] = j; 
c++; 
} 
} 
} 
return res; 
}

#include<iostream>
 
using namespace std;
 
int * addElement(int * array, int * size, int element){
    int * a = new int[*size+1]; //новый массив для того, чтобы вернуть элементы
    for(int i = 0; i < *size; i++)
        a[i] = array[i]; //копируем массив
    a[*size] = element; //добавляем последний элемент
    (*size)++; //увеличиваем счетчик, так как он передан как указатель, то вне функции мы его тоже изменяем
    free(array); //освобождаем память предыдущего массива
    return a; //возвращаем новый массив
}
 
int main(){
 
    int ** a, n, m;
    cout << "Input matrix size [n m]: ";
    cin >> n >> m;
    a = new int*[n];
 
    cout << "Input matrix:" << endl;
    for(int i = 0; i < n; i++){
        a[i] = new int[m];
        for(int j = 0; j < m; j++)
            cin >> a[i][j];
    }
 
    int * result = new int[0], size = 0;
 
    for(int i = 0; i < m; i++) //m и n меняем местами, чтобы проходить по массиву вертикально
        for(int j = 0; j < n; j++)
            if(a[j][i] == 1){
                result = addElement(result, &size, i + 1); //ну или не +1, смотря как вы делаете вывод, с нуля или с единицы
                break;
            }
 
    cout << "Result:" << endl;
    for(int i = 0; i < size; i++)
        cout << result[i] << ' ';
    cout << endl;
 
    return 0;
}

#include <iostream>
#include <ctime>
using namespace std;
 
int main()
{
    srand((int)time(0));
    int n, m, k, p=0, t=0;
    cout <<"Enter n: "; cin >>n;
    cout <<"Enter m: "; cin >>m;
    
      double **a = new double*[n]; 
    for (int i = 0; i < n; i++)
      a[i]=new double[m];
    
    for (int i = 0; i < n; i++)
    {
    for (int j = 0; j < m; j++)
    {
    a[i][j]=rand()%9 + 1;
    cout <<a[i][j]<<" ";
    }
    cout << endl;
    }
    
    for (int j = 0; j < m; j++)
    {
    k=0;    
    for (int i = 0; i < n; i++) 
    if (a[i][j]==1) {k++;break;};
    if (k!=0) p++;
    }
    
    cout <<"OUTPUT:\n";
    if (p==0) cout <<"Cols with 1 are absent\n";
    else
    {
    int*b = new int[p];
    
    for (int j = 0; j < m; j++)
    {
    k=0;    
    for (int i = 0; i < n; i++) 
    if (a[i][j]==1) {k++;break;}
    if (k!=0) b[t++]=j+1;
    }   
    
    for (int i = 0; i < p; i++)
      cout <<b[i]<<" ";
      cout <<endl;
    delete [] b;  
    }  
    for (int i = 0; i < n; i++)
    delete [] a[i];
    delete [] a;    
system("pause");
return 0;
}