biryu2205 / Biryu

0 stars 0 forks source link

Hackerrank Java Instanceof keyword #80

Closed biryu2205 closed 6 years ago

biryu2205 commented 6 years ago
import java.util.ArrayList;
import java.util.Scanner;

class Student {
}

class Rockstar {
}

class Hacker {
}

class InstanceOFTutorial {
  static String count(ArrayList mylist) {
    int a = 0, b = 0, c = 0;
    for (int i = 0; i < mylist.size(); i++) {
      Object element = mylist.get(i);
      if (element instanceof Student) {
        a++;
      }
      if (element instanceof Rockstar) {
        b++;
      }
      if (element instanceof Hacker) {
        c++;
      }
    }
    String ret = Integer.toString(a) + " " + Integer.toString(b) + " " + Integer.toString(c);
    return ret;
  }

  public static void main(String[] args) {
    ArrayList mylist = new ArrayList();
    Scanner sc = new Scanner(System.in);
    int t = sc.nextInt();
    for (int i = 0; i < t; i++) {
      String s = sc.next();
      if (s.equals("Student")) mylist.add(new Student());
      if (s.equals("Rockstar")) mylist.add(new Rockstar());
      if (s.equals("Hacker")) mylist.add(new Hacker());
    }
    System.out.println(count(mylist));
  }
}
letientai299 commented 6 years ago