Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import java.util.*;
- public class SecretSanta {
- public static void main(String[] args) {
- Scanner scanner = new Scanner(System.in);
- int n = scanner.nextInt();
- Map<String, Map<String, Integer>> map = new HashMap<>();
- for(int i = 0; i < n; i++) {
- String komanda = scanner.next();
- if(komanda.equals("wish")) {
- String A = scanner.next(), B = scanner.next();
- map.putIfAbsent(A, new HashMap<>());
- Map<String, Integer> tmp_map = map.get(A);
- tmp_map.put(B, tmp_map.getOrDefault(B, 0) + 1);
- }
- else if(komanda.equals("ask")) {
- String komanda2 = scanner.next();
- if(komanda2.equals("employee")) {
- String ime = scanner.next();
- if(map.containsKey(ime)) {
- System.out.println(map.get(ime).size());
- }
- else {
- System.out.println(0);
- }
- }
- else {
- String A = scanner.next();
- String B = scanner.next();
- int AdoB = 0;
- if(map.containsKey(A) && map.get(A).containsKey(B)) {
- AdoB = map.get(A).get(B);
- }
- int BdoA = 0;
- if(map.containsKey(B) && map.get(B).containsKey(A)) {
- BdoA = map.get(B).get(A);
- }
- System.out.println(AdoB + BdoA);
- }
- }
- }
- }
- }
- /*
- Ana -> Bob, Elena
- Bob -> Ana
- */
Advertisement
Add Comment
Please, Sign In to add comment