【Leetcode】1188. Design Bounded Blocking Queue

题目地址：

https://leetcode.com/problems/design-bounded-blocking-queue/

要求设计一个阻塞队列，实现下面的操作：
1、以一个大于$0$的容量$c$初始化，队列里不能有多于$c$个元素；
2、将某个元素$x$入队；如果入队的时候队列的大小已经达到了$c$，则阻塞住该线程；
3、将队头出队；如果出队的时候队列为空，则阻塞住该线程；
4、返回当前队列里元素个数。

用两个条件变量，和一把锁，两个条件变量一个负责阻塞和唤醒入队线程，另一个负责阻塞和唤醒出队线程。我们不希望入队和出队同时进行，所以需要用同一把锁。当入队的时候，首先上锁（不让出队线程进行），然后等待队列不满，如果不满则入队，并且唤醒出队线程（如果当前有出队线程wait的话，其就会被唤醒做出队操作）；当出队的时候，类似的，首先上锁，然后等待队列不空，如果不空则出队，并唤醒入队线程（如果当前有入队线程wait的话）。求size的时候也要进行上锁。代码如下：

class BoundedBlockingQueue {
 private:
  mutex mu;
  condition_variable cv_full, cv_empty;
  queue<int> q;
  int capacity;

 public:
  BoundedBlockingQueue(int capacity) { this->capacity = capacity; }

  void enqueue(int element) {
    unique_lock<mutex> lk(mu);
    cv_full.wait(lk, [this] { return q.size() < capacity; });
    q.push(element);
    cv_empty.notify_all();
  }

  int dequeue() {
    unique_lock<mutex> lk(mu);
    cv_empty.wait(lk, [this] { return q.size() > 0; });
    int x = q.front();
    q.pop();
    cv_full.notify_all();
    return x;
  }

  int size() {
    lock_guard<mutex> lk(mu);
    return q.size();
  }
};