输出n阶单位矩阵 - C语言

输入正整数n。

输出n阶单位矩阵。

#include<stdio.h>

int main(){
	int n, i, j;
	scanf("%d", &n); 
	
	int a[n][n]; 
	
	 
	for(i=0;i<n;i++){
		for(j=0;j<n;j++){
			if(i==j){
				a[i][j]=1;
			}else{
				a[i][j]=0;
			}
		}
	} 
	
		
	for(i=0;i<n;i++){
		for(j=0;j<n;j++){
			printf("%d ", a[i][j]);
		}
		printf("\n");
	}	
	
	return 0;
}