View Code of Problem 51

#include <stdio.h>
int main(int argc, char const *argv[])
{
    int n;
    scanf("%d", &n);
    int i, j;
    int a[n][n];
    int sum = 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++)
        {
            if (i == j || i - j == 1 || i - j == 2 || i - j == 3 || i - j == 4)
            {
                sum += a[i][j];
            }
        }
    }
    printf("%d\n", sum);

    return 0;
}