Ttren

class Tren{
   Vagon[] v = new Vagon[3];
   private int capacitate_colete;

    public Tren(Vagon[] vagoane){
        v = vagoane;
    }
    public int getNrColete(){
        int s = 0, i;
        for (i = 0; i < v.length; i++)
        s += v[i].getNrColet();
        return s;
    }
     public boolean equals(Object o) {
        if(o instanceof Tren)
            return (((Tren)o).getNrColete() == this.getNrColete());
        else
            return false;
    }
    public String toString(){
        String s="";
        int i;
        for(i = 0; i < v.length; i++)
            s+=v[i].toString() + " ";
        return s;
    }
}

class Vagon{
    private int capacitate_colete;
    public Vagon(int colet)
    {
        capacitate_colete = colet;
    }
    public int getNrColet(){
        return this.capacitate_colete;
    }
}

class Calatori extends Vagon{
    public int nr_pasageri;
    public Calatori(int colet, int pasageri){
        super(colet);
        nr_pasageri = pasageri;
    }


    public void deschideUsa(){
    System.out.println("Atentie!, usile de deschid.");

    }
    public void inchideUsa(){
    System.out.println("Atentie!, usile de inchid.");
    }
    public int getNrPas(){
        return this.nr_pasageri;
    }
}
class CalatoriA extends Calatori{
    private static final int NR_Pasageri = 40;
    private static final int NR_Colete = 300;
    public CalatoriA(){
        super(NR_Pasageri, NR_Colete);
    }
    public String toString(){
        return "CalatoriA 40(pasageri) 300(colete)";
    }
}

class CalatoriB extends Calatori{
    private static final int NR_Pasageri = 50;
    private static final int NR_Colete = 400;
    public CalatoriB(){
        super(NR_Pasageri, NR_Colete);
    }
    public void blocareGeam(){
        System.out.println("Atentie!, geamurile sunt blocate.");
    }
    public String toString(){
        return "CalatoriB 40(pasageri) 300(colete)";
    }
}

class Marfa extends Vagon{
    private static final int NR_Colete = 400;
    public Marfa(){
    super(NR_Colete);
    }
     public String toString(){
        return "Marfa 400(colete)";
    }
}

class Main{
    public static void main(String[] args){
    Vagon[] tren_1 = new Vagon[3];
        tren_1[0] = new Marfa();
        tren_1[1] = new CalatoriA();
        tren_1[2] = new CalatoriB();
      Tren t_1 = new Tren(tren_1);
    Vagon[] tren_2 = new Vagon[3];
        tren_2[0] = new CalatoriA();
        tren_2[1] = new CalatoriB();
        tren_2[2] = new CalatoriB();
      Tren t_2 = new Tren(tren_2);
      System.out.println("in trenul 2: ");
      ((CalatoriB)tren_2[2]).blocareGeam();
      System.out.println("in trenul 1: ");
      ((CalatoriA)tren_1[1]).deschideUsa();
      ((CalatoriA)tren_1[1]).inchideUsa();
        System.out.println("Tren_1");
        System.out.println(t_1);
        System.out.println("Tren_2");
        System.out.println(t_2);
        System.out.println(t_2.equals(t_1));
    }
}