关于使用TreeMap按照value进行排序的解决方案

来源：互联网发布：如何带端口ping 编辑：程序博客网时间：2024/05/17 13:39

今天在写Dijkstra's Algorithm用heap实现的过程中，遇到类似于使用TreeMap按照value进行排序的情况，即每个node对应一个当前最短路，将这些node按照最短路的距离排序，但是同时需要做到随时找到该点并作必要的操作。由于TreeMap是按键值排序并且不能有重复，所以产生了问题，以下提出两个解决办法：

使用TreeMap按照value进行排序

具体的实现过程可以参看该链接：http://blog.csdn.net/liuxiao723846/article/details/50454622

2. 使用优先队列，新建一个私有类存储node和distance，然后按照distance进行排序。

如何实现直接找到某个node对应的私有类呢？这里我们再建立一个数组记录每一个私有类对象的位置，这样就可以随时对某个node的distance做出修改。具体代码如下：

import java.io.IOException;import java.nio.file.Paths;import java.util.ArrayList;import java.util.HashSet;import java.util.PriorityQueue;import java.util.Scanner;import java.util.TreeMap;public class Week2_2 {        private ArrayList<ArrayList<edge>> adjList;//    private TreeMap<Integer,Integer> heap;    private PriorityQueue<nodeDistance> queue;    private nodeDistance[] recordNode;//    private int[] minDistance;    private int n;    private boolean[] x;    private class edge{        private int pointTo;        private int distance;    }        private class nodeDistance implements Comparable<nodeDistance>{        int node;        int distance;                public int compareTo(nodeDistance that){            if(this.distance<that.distance)return -1;            if(this.distance>that.distance)return 1;            return 0;        }    }        public Week2_2(int n) {        this.n=n;//        minDistance=new int[n];//        minDistance[0]=0;        recordNode=new nodeDistance[n];        x=new boolean[n];        queue= new PriorityQueue<nodeDistance>();        adjList=new ArrayList<ArrayList<edge>>();    }        public void readData(String filename) throws IOException{                Scanner in=new Scanner(Paths.get(filename));                for(int i=0;i<n;i++){            String temp=in.nextLine();            String[] list=temp.split("\\s+");//            System.out.println(list[list.length-1]);            ArrayList<edge> adj=new ArrayList<edge>();                        for(int j=1;j<list.length;j++){                edge tempEdge=new edge();                int index=list[j].indexOf(",", 0);                tempEdge.pointTo=Integer.parseInt(list[j].substring(0, index))-1;                tempEdge.distance=Integer.parseInt(list[j].substring(index+1, list[j].length()));                adj.add(tempEdge);            }                        adjList.add(adj);        }    }                public void dijkstraSP(){        //initialize        for(int i=0;i<n;i++){            nodeDistance dis=new nodeDistance();            dis.node=i;            dis.distance=1000000;            recordNode[i]=dis;            queue.add(dis);        }              recordNode[0].distance=0;                        //start the algorithm        while(!queue.isEmpty()){                        nodeDistance nodedistance=queue.poll();            int node=nodedistance.node;            x[node]=true;                        ArrayList<edge> adjNode=adjList.get(node);                        for(int i=0;i<adjNode.size();i++){                int pointToNode=adjNode.get(i).pointTo;                if(!x[pointToNode]){                    queue.remove(recordNode[pointToNode]);                    recordNode[pointToNode].distance=Math.min(recordNode[pointToNode].distance, nodedistance.distance+adjNode.get(i).distance);                    queue.add(recordNode[pointToNode]);                }            }                                }    }        public void OutPut(){        int[] set={7,37,59,82,99,115,133,165,188,197};        HashSet<Integer> requireSet=new HashSet<Integer>();        for(int i=0;i<set.length;i++){            requireSet.add(set[i]-1);        }        for(int i=0;i<n;i++){            if(requireSet.contains(i)){//                System.out.println("Node "+i+"'s distance= "+recordNode[i].distance);                System.out.print(recordNode[i].distance+",");            }        }    }        public static void main(String[] args) throws IOException {        Week2_2 test=new Week2_2(200);        test.readData("./data/dijkstraData.txt");//        test.readData("./data/test.txt");        test.dijkstraSP();        test.OutPut();    }}

阅读全文

0 0