Найти сумму элементов главной диагонали матрицы

#include <stdafx.h>
#include <time.h>
#include <iostream>
#include <conio.h>
 
using namespace std;
void main()
{
 setlocale(LC_ALL, "Russian");
 
 int a, n, sum=0;
 int i, j, A[10][10]={0};
 
 cout << "Будете самостоятельно заполнять массив? 1(да)/0(нет)"<<endl;
 cin >> a;
 
 if (a==1) //1 вариант заполения массива - с клавиатуры
 {
  cout << "Введите размер массива: ";
  cin >> n;
  for(i=0;i<=n-1;i++)
  for(j=0;j<=n-1;j++)
  {
   cout << "A[" << i << "][" << j << "] = ";
   cin >> A[i][j];
  }
 }
 
 else if (a==0) //2 вариант - рандом, но он не работает дальше у словия
 {
  int n=2;
  for(i=0;i<=n-1;i++)
  for(j=0;j<=n-1;j++) 
  A [i][j] = rand()%10; //каждый элемент массива - случайное число от 0 до 10
 }
 
for(i=0;i<=n-1;i++)//вывод массива, работает с клавиатуры, а рандом - не выводит
{
for(j=0;j<=n-1;j++)
 {
  cout <<" "<<A[i][j]<<" ";
 }
  cout<<endl;
}
 
while(j<=n-1 && i<=n-1) //подсчет суммы элементов главной диагонали с помощью цыкла while
{
  if(i=j)
  sum+=A[i][j];
}
 
cout << endl << "Сумма элементов главной диагонали = " << sum << endl;
system("pause");
}

#include <time.h>
#include <iostream>
#include <conio.h>
 
using namespace std;
void main()
{
setlocale(LC_ALL, "Russian");
 
int a, n, sum=0;
int i, j, A[10][10]={0};
 
cout << "Будете самостоятельно заполнять массив? 1(да)/0(нет)"<<endl;
cin >> a;
cout << "Введите размер массива: ";
cin >> n;
if (a==1) //1 вариант заполения массива - с клавиатуры
{
 
for(i=0;i<=n-1;i++)
for(j=0;j<=n-1;j++)
{
cout << "A[" << i << "][" << j << "] = ";
cin >> A[i][j];
}
}
 
else if (a==0) //2 вариант - рандом, но он не работает дальше у словия
{
int n=2;
for(i=0;i<=n-1;i++)
for(j=0;j<=n-1;j++)
A [i][j] = rand()%10; //каждый элемент массива - случайное число от 0 до 10
}
 
for(i=0;i<=n-1;i++)//вывод массива, работает с клавиатуры, а рандом - не выводит
{
for(j=0;j<=n-1;j++)
{
cout <<" "<<A[i][j]<<" ";
}
cout<<endl;
}
j=0;
while(j<n) //подсчет суммы элементов главной диагонали с помощью цыкла while
{
sum+=A[j][j];
j++;
}
 
cout << endl << "Сумма элементов главной диагонали = " << sum << endl;
system("pause");
}

#include "stdafx.h"
#include <time.h>
#include <iostream>
#include <conio.h>
 
using namespace std;
void main()
{
setlocale(LC_ALL, "Russian");
 
int a, n, m, sum=0;
int i, j, A[10][10]={0};
 
cout << "Будете самостоятельно заполнять массив? 1(да)/0(нет)"<<endl;
cin >> a;
 
cout << "Введите размер массива: ";
cin >> n;
 
//1 вариант - заполения массива с клавиатуры
if (a==1) 
{
 for(i=0;i<n;i++)
 for(j=0;j<n;j++)
 {
  cout << "A[" << i << "][" << j << "] = ";
  cin >> A[i][j];
 }
}
 
//2 вариант - рандомное заполнение масива
if (a==0)
{
 srand(time(NULL));// для генерирования случайного числа(используя текущее время)
 for(i=0;i<n;i++) 
 for(j=0;j<n;j++)
 {
  m = 0+rand()%10; // каждый элемент массива - случайное число от 0 до 10
  A[i][j] = m;
 }
} 
 
//вывод массива
for(i=0;i<n;i++)
{
 for(j=0;j<n;j++)
 {
  cout <<" "<<A[i][j]<<" ";
 }
 cout<<endl;
}
 
//подсчет суммы элементов главной диагонали с помощью цыкла while
j=0;
while(j<n) 
 {
 sum+=A[j][j];
 j++;
 }
 
cout << endl << "Сумма элементов главной диагонали = " << sum << endl;
system("pause");
}