if, composição

Fiz a classe aluno, professor, disciplina e o main. Ta funcionando, o problema é q o aluno 1 só pode ter 1 disciplina e o aluno 2 pode ter as 2, no enunciado da questão pedia pra criar duas variáveis de instancia para a classe aluno e foi isso q fiz, o problema está no if do main. Não consigo imprimir 1 disciplina só para o aluno 1. Ta aí o código: public class ControleAcademico {

 static Scanner in = new Scanner(System.in). public static void main(String args[]){ Aluno aluno[]. aluno = new Aluno[2]. for(int i = 0. i < 2. i++){ aluno[i] = new Aluno(). System.out.println("Informe o nome do aluno "+(i+1)). aluno[i].setNome(in.nextLine()). System.out.println("Informe o curso do aluno"+(i+1)). aluno[i].setCurso(in.nextLine()). System.out.println("Informe a disciplina que o aluno "+(i+1)+" cursa"). if(i == 0) aluno[0].setDis1(cadastraDisciplina()[i]). else if(i == 1) aluno[i].setDis1(cadastraDisciplina()[i]). aluno[i].setDis2(cadastraDisciplina()[i]). } for(int k = 0. k < 2. k++){ System.out.println(aluno[k].toString()). } } public static Disciplina[] cadastraDisciplina(){ Disciplina d[]. d = new Disciplina[2]. for(int i = 0. i < 2. i++){ d[i] = new Disciplina(). System.out.println("Informe o nome da disciplina "+(i+1)). d[i].setNome(in.nextLine()). System.out.println("Informe a qual curso a disciplina "+(i+1)+" pertence"). d[i].setCurso(in.nextLine()). System.out.println("Informe o código da disciplina "+(i+1)). d[i].setCodigo(in.nextInt()). in.nextLine(). System.out.println("Informe o professor responsável pela disciplina "+(i+1)). d[i].setProfessor(cadastraProfessor()). } return d. } public static Professor cadastraProfessor(){ Professor p = new Professor(). System.out.println("Informe o nome do professor"). p.setNome(in.nextLine()). System.out.println("Informe o CPF do professor"). p.setCpf(in.nextLine()). System.out.println("Informe a área do professor"). p.setAera(in.nextLine()). System.out.println("Informe o nível de carreira do professor"). p.setNivelCarreira(in.nextInt()). in.nextLine(). return p. } }