функции

[blain]

В матрицах А (3*3) и С (2*2) заменить все элементы, лежащие на главной и побочной диагоналях, единицами.

#include <stdio.h>
#include <conio.h>
# define A 3
# define C 2
void in_mas(int*, int);
void out_mas(int*, int);
void fun(int*, int);
int main(void)
{
    int  a[A][A], c[C][C], k;
    printf("Vvedite mmatr A:\n");
    in_mas(&a[0][0], A*A);
    fun(&a[0][0], k);
    printf(" mmatr A:\n");
    k=A;
    out_mas(&a[0][0], k);
    printf("Vvedite mmatr C:\n");
    k=C;
    in_mas(&c[0][0], C*C);
    fun (&c[0][0], k);
    printf(" mmatr C:\n");
    out_mas(&c[0][0], C*C);
    getch ();
    return 0;
}
 void in_mas(int *p, int b)
{
int i;
for (i=0; i<b; i++)
scanf ("%d\n", p+i);
}

void fun(int *p, int k)
{ 
  int i, j;
  for (i=0; i<k; i++)
  for (j=0; j<k; j++)
   if ((i==j)||(k-i-j==0))  *(p+i)=1;
}
       
void out_mas(int *p, int k)
{
int i, b;
b=k*k;
for (i=0; i<b; i++)
printf ("  %d   ", *(p+i));

}

прога не считает, помогите