迂回方正2

#include<stdio.h>
#include<math.h>
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        f(n);
    }
    return 0;
}
void f(int n){
    int i,j=0,k=1,a[100][100],s=0;
    for(i=0;i<n&&j<n;)
       {
          a[i][j]=k++;
          if(i==0)
            {
                s=j+1;
                i=i+s;
                j=0;
            }
           else if(i!=0&&j<s)
                j=j+1;
           else
               i=i-1;
}
  for(i=0;i<n;i++){
        for(j=0;j<n;j++)
        printf("%5d",a[i][j]);
        printf("\n");
    }

}