c++

[*Fantom*]

нужно переделать прогу чтобы размерность матрицы вводилась при выполнении программы и само решение задачи было оформлено в виде функции, которой передается матрица и ее размерность.
Задание звучит так:
Заполнить матрицу случайными числами. Розбити матрицу на квадраты размером 3х3. В центре каждого квадрата поместить сумму остальных элементов квадрата.
Вот моя прога)

#include <stdio.h>
#include <conio.h>
#include <time.h>
#include <stdlib.h>
#define S 4
double n,m;	
int i,j;
int A[S][S];
int main(void) {
srand(time(0));
for(i=0;i<S;i++){
	for(j=0;j<S;j++){
		A[i][j]=rand()%5;
		printf("%2d",A[i][j]);}
	printf("\n");}
for(i=0;i<S;i++){
	for(j=0;j<S;j++){
		m=(i+2)%3;
		n=(j+2)%3;
		if( n==0 && m==0 )
			A[i][j]=A[i-1][j-1]+A[i-1][j]+A[i-1][j+1]+A[i][j+1]+A[i+1][j+1]+A[i+1][j]+A[i+1][j-1]+A[i][j-1];}}
printf(" poly4ena9 matrica:\n");
for(i=0;i<S;i++){
	for(j=0;j<S;j++){
printf("%3d ",A[i][j]);}
	printf("\n");}
  getch();  
  return 0;}

[challengerr]

Примерно так

#include <stdio.h>
#include <conio.h>
#include <time.h>
#include <stdlib.h>
#include <malloc.h>

void f1(int** m, int sz)
{
 int i, j;
 int a, b;

 for (i = 0; i < sz; i++) 
 {
  for (j = 0; j < sz; j++) 
  {
   m[i][j] = rand()%5; 
   printf ("%d ", m[i][j]); 
  }
  printf("\n"); 
 }

 printf("\n"); 

 for (i = 0; i < sz; i++) 
 {
  for (j = 0; j < sz; j++) 
  {
   a = (i+2)%3; 
   b = (j+2)%3; 
   if (a == 0 && b == 0 && i +1 < sz && j + 1 < sz) 
   {
     
     m[i][j] = m[i-1][j-1] + m[i+1][j+1] + m[i-1][j] + m[i][j-1] + m[i][j+1] + m[i+1][j] + m[i-1][j+1] + m[i+1][j-1];} 
   } 
 }

 for (i = 0; i < sz; i++) 
 {
  for (j = 0; j < sz; j++) 
  {
    printf ("%d ", m[i][j]); 
  } 
  printf("\n"); 
 }

 printf("\n"); 
}

int main()
{
 int i, j, n;
 int** m;
 scanf("%d", &n);
 m = (int**) malloc( n * sizeof(int*) ); for (i=0; i< n; i++) m[i] = (int*) malloc( n * sizeof(int) );
 f1(m, n);

 for (i = 0; i < n; i++) 
 {
  for (j = 0; j < n; j++) 
  {
    printf ("%d ", m[i][j]); 
  } 
  printf("\n"); 
 }

 for (i=0; i< n; i++) free (m[i]); free(m);
 return 0;
}

[*Fantom*]

Спасибо огромное очень помог)