求NxN方阵中主对角元素之和的程序—我写的是3X3的方阵 - 编程论坛

     我学的是C++，所以就用它写了
     #include<iostream>
    using namespace std;
      int i,j;
    int A[3][3]={ 1,2,3,4,5,6,7,8,9 };
    int sum=0;

   int getsum()
   {
   for(i=0; i<=2; i++)
   {
       for(j=0;j<=2;j++)
       {
           if(i==j)
           {   sum=sum+A[i][j];
           }
       }

       }
       return sum;
   }
int main ()
{ int s;
  s=getsum();
  cout<<"the sum is : "<<s<<endl;
  return 0;
}
另提2个问题求：a,有没有更简单的方法。。。b,求非对角元素之和的函数。（用C++）