Возвращение массива из функции

#include <stdio.h>
#include <conio.h>
#include <iostream>
#include <time.h>
#include <stdlib.h>
 
void arr_input(int *arr, int n,int a, int b) {
    for (int i = 0; i < n; i++)
            arr[i]=a+rand()%(b-a);
}  
void arr_output (int *arr, int n) {
    for (int i = 0; i < n; i++)
        printf(" %d ",arr[i]);
}
int arr_count (int *arr, int n) {
    int count_ch = 0;
    for (int i = 0; i < n; i++) {
        int count = 1;
        
        while (count != 0) {
            int ch = arr[i]%10;
            if (ch<0) 
                ch *= -1;
            if (ch == 4) 
                count_ch++;
            count --;
        }
    } 
    return count_ch;
}
void arr_ent (int *arr, int *brr, int n) {
    int j=0;
    for (int i = 0; i < n; i++) {
        int count = 1;
        while (count != 0) {
            int ch = arr[i]%10;
            if (ch<0) 
                ch *= -1;
            if (ch == 4) {
                brr[j]=arr[i];
                j++;
            }
            count --;
        }
    }
    arr_output (brr, j);
}
 
int main() {    
    int *arr, *array, n, min, max, count,res,count_ch;   
    
    printf("Введите n:\n ");
    scanf("%d",&n);//ввод размерности    
    printf("Введите левую границу промежутка:\n ");
    scanf("%d",&min);//ввод границы начала
    printf("Введите правую границу промежутка:\n ");
    scanf("%d",&max);//ввод границы конца
    system("cls");
 
    srand(time(NULL));
    arr = (int *)malloc(n*sizeof(int));
    arr_input(arr,n,min,max);
    
    printf("Исходный массив: \n");
    arr_output(arr,n);
    
    count = arr_count(arr,n);
    
    if (count == 0) {
        printf("\n\nВыходящий массив пуст.");
    } 
    else {
        array=(int *)malloc(sizeof(count));
        printf("\n\nВыходящий массив:\n");
        
        arr_ent(arr,array,n);      
       
    }
    free(arr);  
    free(array);
    getch();
    return 0;
}