Dijkstra的算法 - 优先级队列问题

时间:2012-11-27 22:55:00

标签: java priority-queue dijkstra

我有一个带有一堆节点,边缘等的Graph类,我正在尝试执行Dijkstra的算法。我开始将所有节点添加到优先级队列。每个节点都有一个布尔标志,表示它是否已经“已知”,对它前面的节点的引用,以及一个存储源节点长度的int dist字段。在将所有节点添加到PQ然后适当地标记源节点之后,我注意到错误的节点首先被从PQ中拉出。应该是具有最小dist字段的节点首先脱离(因为它们都被初始化为非常高的数字,除了源,PQ的第一个节点应该是源...除了它不是为某些原因)。 下面是我的Node类中我的compare方法后面的算法代码。

    public void dijkstra() throws IOException {
    buildGraph_u();
    PriorityQueue<Node> pq = new PriorityQueue<>(200, new Node());

    for (int y = 0; y < input.size(); y++) {
        Node v = input.get(array.get(y));
        v.dist = 99999;
        v.known = false;
        v.prnode = null;
        pq.add(v);
    }

    source.dist = 0;
    source.known = true;
    source.prnode = null;
    int c=1;
    while(c != input.size()) {
        Node v = pq.remove();
        //System.out.println(v.name);
                    //^ Prints a node that isn't the source
        v.known = true;
        c++;
        List<Edge> listOfEdges = getAdjacent(v);
        for (int x = 0; x < listOfEdges.size(); x++) {
            Edge edge = listOfEdges.get(x);
            Node w = edge.to;
            if (!w.known) {
                int cvw = edge.weight;
                if (v.dist + cvw < w.dist) {
                    w.dist = v.dist + cvw;
                    w.prnode = v;
                }
            }
        }
    }
}

public int compare (Node d1, Node d2) {
       int dist1 = d1.dist;
       int dist2 = d2.dist;
       if (dist1 > dist2) 
         return 1;
       else if (dist1 < dist2) 
         return -1;
       else 
         return 0;
     }

任何人都可以帮我找到PQ的问题吗?

1 个答案:

答案 0 :(得分:2)

优先级队列使用假设,即在插入元素后订单不会更改。

因此,您可以:

,而不是将所有元素都插入优先级队列
  1. 从一个节点开始。

  2. 优先级队列不为空时循环。

  3. 如果元素是“已知”,则不执行任何操作。

  4. 每当您找到较小的距离时,请将其添加到具有“正确”权重的优先级队列中。

  5. 所以你需要在优先队列中存储一个其他的,一对:插入时的距离,节点本身。