电话簿编程-TreeSet或HashSet

Question

我有一个很好的问题——创建一个包含联系人列表的电话簿。正如电话簿所说，

>

联系人总是要分类的。（按姓名）

可以用星号标记某些联系人，因此他们必须高于其他联系人。（按联系人创建时间排序*联系人）

class PhoneBook{
//require an always sorted d.s
TreeSet<Contact> contacts = new TreeSet<Contact>();

@Override
public String toString() {
    return "PhoneBook [contacts=" + contacts + "]";
}

public boolean addContact(Contact contact){
    //validate before adding the contact.
    return contacts.add(contact);
}

}

class Contact implements Comparable<Contact>{
String name;
int phoneNo;
Date timeAdded;
boolean starContact;



public Contact(String name, int phoneNo, Date timeAdded, boolean starContact) {
    super();
    this.name = name;
    this.phoneNo = phoneNo;
    this.timeAdded = timeAdded;
    this.starContact = starContact;
}



@Override
public int compareTo(Contact otherContact) {
    if(this.starContact && otherContact.starContact){
        return this.timeAdded.before(otherContact.timeAdded)?-1:1; //impossible to add 2 contacts at the same time
    }else if(this.starContact){
        return -1;
    }else if(otherContact.starContact){
        return 1;
    }else{
        //simple Contacts
        return this.name.compareTo(otherContact.name);
    }
}






@Override
public String toString() {
    return "\nContact [name=" + name + ", timeAdded=" + timeAdded
            + ", starContact=" + starContact + "]";
}


}

测试代码

    public class MobilePhoneBookDemo {

/**
 * @param args
 */
public static void main(String[] args) {

    PhoneBook phoneBook = new PhoneBook();

    Contact frnd1 = new Contact("Z",56,new Date(),false);
    phoneBook.addContact(frnd1);
    Contact frnd2 = new Contact("A",3,new Date(),false);
    phoneBook.addContact(frnd2);
    Contact frnd3 = new Contact("C",30,new Date(),false);
    phoneBook.addContact(frnd3);
    try {
        TimeUnit.SECONDS.sleep(1);
    } catch (InterruptedException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    }
    Contact ta = new Contact("Ta", 5, new Date(), true);
    phoneBook.addContact(ta);
    Contact ma = new Contact("Ma", 31, new Date(), true);
    phoneBook.addContact(ma);
    Contact baba = new Contact("Baba", 300, new Date(), true);
    phoneBook.addContact(baba);

    //change the priority later for one of my friends.
    System.out.println(phoneBook);
    frnd1.starContact = true;
    System.out.println(phoneBook.contacts.contains(frnd1));

    if(phoneBook.contacts.remove(frnd1)){
        System.out.println("removed");
        phoneBook.contacts.add(frnd1);
    }

    System.out.println(phoneBook);
}

}

面临的问题：

1. 容器再也找不到条目了，怎么了？我确实试过在Contact上加上equals和hashcode，显然，如果存在一个Comparator/Comparable，那么compare*只会被调用

问题按顺序排列。我很感激所有的答案，但这确实是一个完整的测试用例，所以请在提供答案之前尝试运行PhoneBook一次。谢谢。

Answer 1

解决方案很简单，首先删除，然后更改值，然后再次添加。我首先对修改进行评论，我还加入了修改后的compareTo（），因为它有很多混乱之处。

//frnd1.starContact = true;
    System.out.println("Entry present>"+phoneBook.contacts.contains(frnd1));

    if(phoneBook.contacts.remove(frnd1)){
        System.out.println("removed");
        frnd1.starContact = true;
        phoneBook.contacts.add(frnd1);
    }


@Override
public int compareTo(Contact otherContact) {
    if(otherContact.phoneNo == this.phoneNo){
        return 0;
    }
    if(this.starContact && otherContact.starContact){
        return this.timeAdded.before(otherContact.timeAdded)?-1:1; //impossible to add 2 contacts at the same time
    }else if(this.starContact){
        return -1;
    }else if(otherContact.starContact){
        return 1;
    }else{
        //simple Contacts
        return this.name.compareTo(otherContact.name);
    }
}

Answer 2

你让我在

联系人总是要分类的。（按姓名）

如果顺序很重要，请使用TreeSet。元素存储在各种二叉查找树中，这意味着默认情况下它们是排序的。

另一方面，HashSet不保证任何顺序、排序甚至插入顺序。

编辑——试着用这种方式写下你的条件，更具可读性，也更健壮。条件的顺序就是比较的优先顺序。

if(a != b){
    return a<b;
}

if(c != d) {
    return c<d;
}

//and so on.

return 0;

Answer 3

这一行：

返回this.timeAdded.before（其他Contact.time添加）？-1:1；

如果您将联系人与其自身进行比较，则永远不会返回0。因此，集合将无法找到包含（）的对象