Codigo fuente que Llena con 1 la diagonal principal del matriz en java

Codigo fuente que Llena con 1 la diagonal principal del matriz en java

public class Diagonal {

    /**
     * @param args the command line arguments
     */
    public static void main(String[] args) {
        // TODO code application logic here
        int a[][] = new int[5][5];
        int i=0;
        int j=0;
       
      //  System.out.println(a.length);
       
        for(i=0;i<a.length;i++){
            for(j=0;j<a.length;j++){
                if(i==j){
                a[i][j]=1;
                }
                else{
                a[i][j]=0;
                   
                }
            }
        }
       
         for(i=0;i<a.length;i++){
            for(j=0;j<a.length;j++){
     
               System.out.print(a[i][j]);
               if(j==a.length-1){
                System.out.print("\n");
               }
            }
         }
       
}
    }