Josif_tepe

Untitled

Jan 9th, 2026
30
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 1.73 KB | None | 0 0
  1. import java.util.*;
  2.  
  3. public class SecretSanta {
  4.     public static void main(String[] args) {
  5.         Scanner scanner = new Scanner(System.in);
  6.         int n = scanner.nextInt();
  7.  
  8.  
  9.         Map<String, Map<String, Integer>> map = new HashMap<>();
  10.  
  11.         for(int i = 0; i < n; i++) {
  12.             String komanda = scanner.next();
  13.             if(komanda.equals("wish")) {
  14.                 String A = scanner.next(), B = scanner.next();
  15.  
  16.                 map.putIfAbsent(A, new HashMap<>());
  17.  
  18.                 Map<String, Integer> tmp_map = map.get(A);
  19.                 tmp_map.put(B, tmp_map.getOrDefault(B, 0) + 1);
  20.  
  21.             }
  22.             else if(komanda.equals("ask")) {
  23.                 String komanda2 = scanner.next();
  24.                 if(komanda2.equals("employee")) {
  25.                     String ime = scanner.next();
  26.                     if(map.containsKey(ime)) {
  27.                         System.out.println(map.get(ime).size());
  28.                     }
  29.                     else {
  30.                         System.out.println(0);
  31.                     }
  32.                 }
  33.                 else {
  34.                     String A = scanner.next();
  35.                     String B = scanner.next();
  36.  
  37.                     int AdoB = 0;
  38.                     if(map.containsKey(A) && map.get(A).containsKey(B)) {
  39.                         AdoB = map.get(A).get(B);
  40.                     }
  41.  
  42.                     int BdoA = 0;
  43.                     if(map.containsKey(B) && map.get(B).containsKey(A)) {
  44.                         BdoA = map.get(B).get(A);
  45.                     }
  46.  
  47.                     System.out.println(AdoB + BdoA);
  48.                 }
  49.  
  50.             }
  51.            
  52.         }
  53.     }
  54. }
  55.  
  56. /*
  57.  
  58. Ana -> Bob, Elena
  59.  
  60. Bob -> Ana
  61.  
  62.  
  63. */
  64.  
Advertisement
Add Comment
Please, Sign In to add comment