Matrica - SP Lab 5 100/100

/*Матрица Problem 2 (5 / 14)
Од стандарден влез се читаат димензии на квадратна матрица (макс 100х100) која е составена од целобројни елементи. Во продолжение се читаат сите елементи од матрицата соодветно.

Да се состави нова матрицаво која секој елемент од главната дијагонала ќе биде заменет со сумата на сите елементи од истиот ред десно од елементот и елементите кои се во истата колона над елементот.

Исто така потребно е секој елемент од споредната дијагонала да се замени со сумата на елементи од иститот ред пред елементот и елементите кои се во истата колона под елементот.

Доколку нема соодветни елементи, потребно е елементот да се постави на 0. Доколку некој елемент се наоѓа и на главна и на споредна дијагонала, потребно е двете суми да се сумираат. Елементите кои не се наоѓаат на некоја од дијагоналите не треба да се променат.*/


#include<stdio.h>
#define MAX 100

int main()
{
    int n,i,j;
    scanf("%d",&n);
    int a[n][n];
    int b[MAX][MAX]={0};
    for (i=0;i<n;i++)
        {
            for (j=0;j<n;j++)
            {
                scanf("%d",&a[i][j]);
            }
        }
    for (i=0;i<n;i++)
    {
        for (j=0;j<n;j++)
        {
            int zbir=0;
            if (i==j) //glavna dijagonala
            {
                int z=j+1; //z-elementot desno od elemntot na glavna dijagonala
                while (z<n)
                {
                    zbir+=a[i][z];
                    z++;
                }
                z=i-1; //elementot nad elementot od glavna dijagonala
                while (z>=0)
                    {
                        zbir+=a[z][j];
                        z--;
                    }
                b[i][j]+=zbir;
            }
            if (i+j==n-1) //sporedna dijagonala
                {
                    zbir=0;
                    int z=j-1; //elementot pred elementot na sporedna dijagonala
                    while(z>=0)
                    {
                         zbir+=a[i][z];
                         z--;
                    }
                z=i+1; //elementot pod elementot na sporedna dijagonala
                while(z<n)
                    {
                        zbir+=a[z][j];
                        z++;
                    }
                b[i][j]+=zbir;
            }
            if (i!=j&&((i+j)!=(n-1))) //ako elementot ne pripagja ni na glavna ni na sporedna dijagonala
            {
                b[i][j]=a[i][j];
            }
        }
    }

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

    return 0;

}