hello-algo/ja/codes/cpp/chapter_stack_and_queue/array_deque.cpp

/**
 * File: array_deque.cpp
 * Created Time: 2023-03-02
 * Author: krahets (krahets@163.com)
 */

#include "../utils/common.hpp"

/* 循環配列ベースの両端キュー */
class ArrayDeque {
  private:
    vector<int> nums; // 両端キューの要素を格納する配列
    int front;        // 先頭ポインタ。先頭要素を指す
    int queSize;      // 両端キューの長さ

  public:
    /* コンストラクタ */
    ArrayDeque(int capacity) {
        nums.resize(capacity);
        front = queSize = 0;
    }

    /* 両端キューの容量を取得 */
    int capacity() {
        return nums.size();
    }

    /* 両端キューの長さを取得 */
    int size() {
        return queSize;
    }

    /* 両端キューが空かどうかを判定 */
    bool isEmpty() {
        return queSize == 0;
    }

    /* 循環配列のインデックスを計算 */
    int index(int i) {
        // 剰余演算により配列の先頭と末尾をつなげる
        // i が配列の末尾を越えたら先頭に戻る
        // i が配列の先頭を越えて前に出たら末尾に戻る
        return (i + capacity()) % capacity();
    }

    /* キュー先頭にエンキュー */
    void pushFirst(int num) {
        if (queSize == capacity()) {
            cout << "両端キューがいっぱいです" << endl;
            return;
        }
        // 先頭ポインタを左に 1 つ移動する
        // 剰余演算により、front が配列先頭を越えた後に末尾へ戻るようにする
        front = index(front - 1);
        // num をキュー先頭に追加
        nums[front] = num;
        queSize++;
    }

    /* キュー末尾にエンキュー */
    void pushLast(int num) {
        if (queSize == capacity()) {
            cout << "両端キューがいっぱいです" << endl;
            return;
        }
        // キュー末尾ポインタを計算し、末尾インデックス + 1 を指す
        int rear = index(front + queSize);
        // num をキュー末尾に追加
        nums[rear] = num;
        queSize++;
    }

    /* キュー先頭からデキュー */
    int popFirst() {
        int num = peekFirst();
        // 先頭ポインタを 1 つ後ろへ進める
        front = index(front + 1);
        queSize--;
        return num;
    }

    /* キュー末尾からデキュー */
    int popLast() {
        int num = peekLast();
        queSize--;
        return num;
    }

    /* キュー先頭の要素にアクセス */
    int peekFirst() {
        if (isEmpty())
            throw out_of_range("両端キューが空です");
        return nums[front];
    }

    /* キュー末尾の要素にアクセス */
    int peekLast() {
        if (isEmpty())
            throw out_of_range("両端キューが空です");
        // 末尾要素のインデックスを計算
        int last = index(front + queSize - 1);
        return nums[last];
    }

    /* 出力用の配列を返す */
    vector<int> toVector() {
        // 有効長の範囲内のリスト要素のみを変換
        vector<int> res(queSize);
        for (int i = 0, j = front; i < queSize; i++, j++) {
            res[i] = nums[index(j)];
        }
        return res;
    }
};

/* Driver Code */
int main() {
    /* 両端キューを初期化 */
    ArrayDeque *deque = new ArrayDeque(10);
    deque->pushLast(3);
    deque->pushLast(2);
    deque->pushLast(5);
    cout << "両端キュー deque = ";
    printVector(deque->toVector());

    /* 要素にアクセス */
    int peekFirst = deque->peekFirst();
    cout << "先頭要素 peekFirst = " << peekFirst << endl;
    int peekLast = deque->peekLast();
    cout << "末尾要素 peekLast = " << peekLast << endl;

    /* 要素をエンキュー */
    deque->pushLast(4);
    cout << "要素 4 を末尾に追加した後 deque = ";
    printVector(deque->toVector());
    deque->pushFirst(1);
    cout << "要素 1 を先頭に追加した後 deque = ";
    printVector(deque->toVector());

    /* 要素をデキュー */
    int popLast = deque->popLast();
    cout << "末尾から取り出した要素 = " << popLast << "、末尾から取り出した後 deque = ";
    printVector(deque->toVector());
    int popFirst = deque->popFirst();
    cout << "先頭から取り出した要素 = " << popFirst << "、先頭から取り出した後 deque = ";
    printVector(deque->toVector());

    /* 両端キューの長さを取得 */
    int size = deque->size();
    cout << "両端キューの長さ size = " << size << endl;

    /* 両端キューが空かどうかを判定 */
    bool isEmpty = deque->isEmpty();
    cout << "両端キューが空かどうか = " << boolalpha << isEmpty << endl;
    return 0;
}