View Code of Problem 7

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;


public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int T=scanner.nextInt();
        for (int i = 0; i < T; i++) {
            int n=scanner.nextInt();
            
            Map<Integer, Integer> map1 = new HashMap<>();
            Map<Integer, Integer> map2= new HashMap<>();
            for (int j = 0; j < n; j++) {
                String s=scanner.next();
                String[] split = s.split("=");
                Integer key= Integer.valueOf(split[0]);
                Integer value= Integer.valueOf(split[1]);
                map1.put(key,value);
                map2.put(value,key);
            }
            int m=scanner.nextInt();
            for (int j = 0; j < m; j++) {
                int temp=scanner.nextInt();
                Integer result=map1.get(temp);
                if (result != null) {
                    System.out.println(result);
                } else {
                    result=map2.get(temp);
                    if (result != null) {
                        System.out.println(result);
                    } else {
                        System.out.println("UNKNOW");
                    }
                }
            }
            System.out.println();
        }
    }

}