這幾天一直在搞關於優先順序隊列的實現,因為要考慮到線程的安全,所以PriorityQueue就不適用了。經朋友的指點,終於想到一個非常簡單的實現方法,那就是把優先順序比較好的插入一個隊列,優先順序低的插入另一個隊列,取數的時候先在優先順序高的隊列上取數。這有個缺點就是如果優先順序別越多的話,隊列就越多。 因為要安全執行緒,隊列採用ConcurrentLinkedQueue這個安全執行緒的,而api文檔上說ConcurrentLinkedQueue採用了有效“無等待 (wait-free)”演算法,所以它的輸送量是很不錯的! 簡單代碼如下:
- package test;
- import java.util.concurrent.ConcurrentLinkedQueue;
- public class PriorityQueueTest {
- public static void main(String[] args) {
- ConcurrentLinkedQueue<String> highPriority = new ConcurrentLinkedQueue<String>(); //高優先順序
- ConcurrentLinkedQueue<String> lowPriority = new ConcurrentLinkedQueue<String>(); //低優先順序
-
- highPriority.add("aaa");
- highPriority.add("bbb");
- highPriority.add("111");
-
- lowPriority.add("ccc");
- lowPriority.add("ddd");
- lowPriority.add("222");
-
- int i = 0 ,j = 0, k=0;
- while(true){
- while(true){
- if(!highPriority.isEmpty()){
- System.out.print(highPriority.remove());
- i++;
- k++;
- System.out.println(", i = "+i+", k="+k);
- break;
- }
- if(!lowPriority.isEmpty()){
- System.out.print(lowPriority.remove());
- j++;
- k++;
- System.out.println(", j = "+j+", k="+k);
- break;
- }
- break;
- }
- try {
- Thread.sleep(100);
- } catch (InterruptedException e) {
- e.printStackTrace();
- }
- }
- }
- }
結果: 還有一種是,通過繼承PriorityQueue並實現Comparable介面,然後自已重寫過compareTo方法就能實現很強大的優先順序隊列了,不過缺點是線程不安全的! 代碼如下:
- package test;
- import java.util.PriorityQueue;
- public class PriorityTest extends PriorityQueue<PriorityTest.Test>{
- static class Test implements Comparable<Test>{
- String packet;
- int priotity;
-
- public Test(String packet, int priotity) {
- this.packet = packet;
- this.priotity = priotity;
- }
-
- public int compareTo(Test arg) {
- if(priotity < arg.priotity)
- return 1;
- else if(priotity > arg.priotity)
- return -1;
- else
- return 0;
- }
-
- public String toString(){
- return packet;
- }
- }
-
- public void add(String str, int priority){
- super.add(new Test(str,priority));
- }
-
- public static void main(String args[]){
- PriorityTest pTest = new PriorityTest();
- pTest.add("aaa",3); //優先順序最高
- pTest.add("bbb",2);
- pTest.add("ccc",1);
-
- while(!pTest.isEmpty()){
- System.out.println(pTest.remove());
- }
- }
- }