编写一个函数，计算三行三列矩阵对角线上元素之和。运行出来数字不对，不知道哪里出错了
#include<stdio.h>
int main()
{
int s=0,i,j;
printf("请输入需要的三维数组,以空格隔开\n");
int a[2][2];
for(i=0;i<3;i++)
for(j=0;j<3;j++)
{scanf("%d",&a[i][j]);
if (i==j||i+j==2)
s=s+a[i][j];
}
printf("对角线上元素之和为%d\n",s);
return 0;
}

谢谢了，我这里还有一程序编不出来，不知道哪里出错了
输入一批数，输出比相邻数大的数，当输入0时结束。
#include<stdio.h>
int main()
{
int  i[20],k,j;
scanf("%d",&i[1]);
for (j=1;i[j]!=0;j++)
{
scanf("%d",&i[j]);
for (k=1; k<j;k++)
{if (i[j]<i[k])
break;
}
printf("%d\n",i[j]);
}
return 0;
}