Scala数据结构——优先级队列

优先级队列

思路

上篇介绍的栈和队列无法按照人们想要的顺序对数据进行读取，优先级队列即可按照一定的优先
顺序实现对队列中的数据存取操作。实现原理还是在队列中插入数据时，需要和已知有序的队列
比较元素的大小，然后给当前插入的元素寻找到合适的位置插入即可实现有顺序的队列。因为和
普通队列相似，只不过时优先级队列添加元素的逻辑略有不同，就不画图了。代码见下图：

代码

import util.control.Breaks._

import scala.io.StdIn

object FirstArrayQueue {
  def main(args: Array[String]): Unit = {
    val arrayQue = new FirstArrayQue(5)
    var key = ""

    while (true) {
      println("show:表示显示队列")
      println("exit:表示退出队列")
      println("add:表示添加元素")
      println("get:表示取数据")
      key = StdIn.readLine()
      key match {
        case "show" => arrayQue.showQueue()
        case "exit" => System.exit(0)
        case "add" => {
          println("请输入一个要添加的数字：")
          val value = StdIn.readInt()
          arrayQue.addQueue(value)
        }
        case "get" => {
          val res = arrayQue.getQueue()
          if (res.isInstanceOf[Exception]) {
            println(res.asInstanceOf[Exception].getMessage)
          }
          println(res)
        }
      }
    }
  }

  class FirstArrayQue(arrMaxSize: Int) {
    val maxSize = arrMaxSize
    val arr = new Array[Int](maxSize)
    var rear = -1

    //判断队列是否满
    def isFull(): Boolean = {
      rear == maxSize - 1
    }

    //判断队列是否空
    def isNull(): Boolean = {
      rear + 1 == 0
    }


    //添加元素
    def addQueue(n: Int) = {
      if (isFull()) {
        //判断队列是否满，满了就不能添加元素
        println("队列已满")
      }else{
        //判断队列是否为空，如果为空，那么插入的元素直接插入即可有序
        if (rear == (-1)) {
          arr(rear + 1) = n
          rear += 1
        } else {
          breakable {
            /*不为空说明队列中存在元素，需要拿带插入的元素和已存在的比较（注意：从队头开始比，
            已存在的队列是有序的），通过比较，找到待插入元素的位置即可*/
            for (i <- rear to 0 by -1) {
              if (n > arr(i)) {
                //待插入元素比队列中已知元素大的位置
                arr(i + 1) = arr(i)
              } else {
                //队列元素都比待插入元素大
                arr(i + 1) = n
                rear += 1
                break()
              }
              arr(i) = n
              rear += 1
            }
          }
        }
      }

    }
    //获取数据
    def getQueue(): Any = {
      if (isNull()) {
        return new Exception("队列为空")
      }
      return arr(rear)
    }

    //显示队列的所有数据
    def showQueue(): Unit = {
      if (isNull()) {
        println("queue is empty")
        return
      }
      for (i <- 0 to rear) {
        printf("arr[%d]=%d\n", i, arr(i))

      }
    }

  }

}