I am trying to get acquainted with Collections. I have a String, which is my key, email address and Person object (firstName, lastName, phone, email). I read in the chapter of Java collections on Sun's web pages that if you have a HashMap and you want it to be sorted, you could use TreeMap. How does this view work? Is it based on the compareTo () method that you have in your Person class? I tried the compareTo () method in the Person class to sort by lastName. But it does not work properly and wondered if I have the right idea or not. getSortedListByLastName at the bottom of this code, where I am trying to convert to TreeMap. Also, if this is the right way to do this, or one of the right ways to do this, as I then sort firstName,since comparing compareTo () by lastName.
import java.util.*;
public class OrganizeThis
{
public void add(Person p)
{
staff.put(p.getEmail(), p);
}
public void remove(String email)
{
staff.remove(email);
}
public void empty()
{
staff.clear();
}
public Person findByEmail(String email)
{
Person aPerson = staff.get(email);
return aPerson;
}
public Person[] find(String lastName)
{
ArrayList<Person> names = new ArrayList<Person>();
for (Person s : staff.values())
{
if (s.getLastName() == lastName) {
names.add(s);
}
}
Person nameArray[] = new Person[names.size()];
names.toArray(nameArray);
return nameArray;
}
public Person[] getSortedListByLastName()
{
Map<String, Person> sorted = new TreeMap<String, Person>(staff);
ArrayList<Person> sortedArrayList = new ArrayList<Person>();
for (Person s: sorted.values()) {
sortedArrayList.add(s);
}
Person sortedArray[] = new Person[sortedArrayList.size()];
sortedArrayList.toArray(sortedArray);
return sortedArray;
}
private Map<String, Person> staff = new HashMap<String, Person>();
public static void main(String[] args)
{
OrganizeThis testObj = new OrganizeThis();
Person person1 = new Person("J", "W", "111-222-3333", "JW@ucsd.edu");
Person person2 = new Person("K", "W", "345-678-9999", "KW@ucsd.edu");
Person person3 = new Person("Phoebe", "Wang", "322-111-3333", "phoebe@ucsd.edu");
Person person4 = new Person("Nermal", "Johnson", "322-342-5555", "nermal@ucsd.edu");
Person person5 = new Person("Apple", "Banana", "123-456-1111", "apple@ucsd.edu");
testObj.add(person1);
testObj.add(person2);
testObj.add(person3);
testObj.add(person4);
testObj.add(person5);
System.out.println(testObj.findByEmail("JW@ucsd.edu"));
System.out.println("------------" + '\n');
Person a[] = testObj.find("W");
for (Person p : a)
System.out.println(p);
System.out.println("------------" + '\n');
a = testObj.find("W");
for (Person p : a)
System.out.println(p);
System.out.println("SORTED" + '\n');
a = testObj.getSortedListByLastName();
for (Person b : a) {
System.out.println(b);
}
}
}
:
public class Person implements Comparable
{
String firstName;
String lastName;
String telephone;
String email;
public Person()
{
firstName = "";
lastName = "";
telephone = "";
email = "";
}
public Person(String firstName)
{
this.firstName = firstName;
}
public Person(String firstName, String lastName, String telephone, String email)
{
this.firstName = firstName;
this.lastName = lastName;
this.telephone = telephone;
this.email = email;
}
public String getFirstName()
{
return firstName;
}
public void setFirstName(String firstName)
{
this.firstName = firstName;
}
public String getLastName()
{
return lastName;
}
public void setLastName(String lastName)
{
this.lastName = lastName;
}
public String getTelephone()
{
return telephone;
}
public void setTelephone(String telephone)
{
this.telephone = telephone;
}
public String getEmail()
{
return email;
}
public void setEmail(String email)
{
this.email = email;
}
public int compareTo(Object o)
{
String s1 = this.lastName + this.firstName;
String s2 = ((Person) o).lastName + ((Person) o).firstName;
return s1.compareTo(s2);
}
public boolean equals(Object otherObject)
{
if (this == otherObject) {
return true;
}
if (otherObject == null) {
return false;
}
if (!(otherObject instanceof Person)) {
return false;
}
Person other = (Person) otherObject;
return firstName.equals(other.firstName) && lastName.equals(other.lastName) &&
telephone.equals(other.telephone) && email.equals(other.email);
}
public int hashCode()
{
return this.email.toLowerCase().hashCode();
}
public String toString()
{
return getClass().getName() + "[firstName = " + firstName + '\n'
+ "lastName = " + lastName + '\n'
+ "telephone = " + telephone + '\n'
+ "email = " + email + "]";
}
}