编写一个求n*n方阵对角元素之和及非对角元素之和的函数,并设计一个测试主函数