编一个程序，定义一个n行n列的二维整数数组，赋初值，然后求出对角线上的元素之和。