/** * Created by leon on 1/23/18. */ import java.util. *; public class PhoneBook { TreeMap phoneBook = new TreeMap(); public String add(String name, String number) { phoneBook.put(name, number); return name + " " + number; } public String remove(String name) { phoneBook.remove(name); return name; } public String lookup(String name) { return phoneBook.get(name); } public String reverseLookup(String number) { for(Map.Entry entry : phoneBook.entrySet()) { String value = entry.getValue(); if(value.equals(number)) { return entry.getKey(); } } return ""; } public void display() { for(Map.Entry entry : phoneBook.entrySet()) { String value = entry.getValue(); System.out.println(entry.getKey() + " -> " + entry.getValue()); } } }