如何从优先级队列中删除特定元素?

时间:2012-02-06 00:33:38

标签: java priority-queue

我使用Java API创建了一个优先级队列,我想从程序末尾的优先级队列中删除一个特定元素。我知道它必须对比较器做些什么,但我无法弄明白。有人可以帮忙吗?这是我的代码:

public static void main(String[] args)
{
    PriorityQueue<Element> X = new PriorityQueue<Element>(100, new ElementComparator());
    X.add(new Element(30, 3));
    X.add(new Element(700, 4.5));
    X.add(new Element(100, 6.2));
    X.add(new Element(2, 8.1));
    System.out.println(X.remove(new Element(100, 6.2)));
}

这是我的Element类:

private int index;
private double value;

public Element(int i, double v) 
{
    index = i;
    value = v;
}

public int getIndex() { return index;};
public double getValue() { return value;};
public void setValue(double v) { value = v;};

这是我创建的比较器:

public int compare(Element o1, Element o2)
{
    int idx1 = o1.getIndex();
    int idx2 = o2.getIndex();
    if (idx1 < idx2) {
        return -1;
    } else if (idx1 > idx2) {
        return 1;
    } else {
        return 0;
    }
}

public boolean equals(Element o1, Element o2) 
{
    return o1.getIndex() == o2.getIndex();
}

感谢您的帮助......

1 个答案:

答案 0 :(得分:1)

您需要在Element对象上定义equals()和hashcode():

public class Element{
    private int index;
    private double value;

    public Element(int i, double v)
    {
        index = i;
        value = v;
    }

    public int getIndex() { return index;}
    public double getValue() { return value;}
    public void setValue(double v) { value = v;}

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof Element)) return false;

        Element element = (Element) o;

        if (index != element.index) return false;

        return true;
    }

    @Override
    public int hashCode() {
        return index;
    }
}

在ElementComparator上定义equals()不会执行相同的任务。