Question

1117. H2O 生成

English Version

题目描述

现在有两种线程，氧 oxygen 和氢 hydrogen，你的目标是组织这两种线程来产生水分子。

存在一个屏障（barrier）使得每个线程必须等候直到一个完整水分子能够被产生出来。

氢和氧线程会被分别给予 releaseHydrogen 和 releaseOxygen 方法来允许它们突破屏障。

这些线程应该三三成组突破屏障并能立即组合产生一个水分子。

你必须保证产生一个水分子所需线程的结合必须发生在下一个水分子产生之前。

换句话说:

• 如果一个氧线程到达屏障时没有氢线程到达，它必须等候直到两个氢线程到达。
• 如果一个氢线程到达屏障时没有其它线程到达，它必须等候直到一个氧线程和另一个氢线程到达。

书写满足这些限制条件的氢、氧线程同步代码。

 

示例 1:

输入: water = "HOH"
输出: "HHO"
解释: "HOH" 和 "OHH" 依然都是有效解。

示例 2:

输入: water = "OOHHHH"
输出: "HHOHHO"
解释: "HOHHHO", "OHHHHO", "HHOHOH", "HOHHOH", "OHHHOH", "HHOOHH", "HOHOHH" 和 "OHHOHH" 依然都是有效解。

 

提示：

• 3 * n == water.length
• 1 <= n <= 20
• water[i] == 'O' or 'H'
• 输入字符串 water 中的 'H' 总数将会是 2 * n 。
• 输入字符串 water 中的 'O' 总数将会是 n 。

解法

方法一

from threading import Semaphore


class H2O:
  def __init__(self):
    self.h = Semaphore(2)
    self.o = Semaphore(0)

  def hydrogen(self, releaseHydrogen: "Callable[[], None]") -> None:
    self.h.acquire()
    # releaseHydrogen() outputs "H". Do not change or remove this line.
    releaseHydrogen()
    if self.h._value == 0:
      self.o.release()

  def oxygen(self, releaseOxygen: "Callable[[], None]") -> None:
    self.o.acquire()
    # releaseOxygen() outputs "O". Do not change or remove this line.
    releaseOxygen()
    self.h.release(2)

class H2O {
  private Semaphore h = new Semaphore(2);
  private Semaphore o = new Semaphore(0);

  public H2O() {
  }

  public void hydrogen(Runnable releaseHydrogen) throws InterruptedException {
    h.acquire();
    // releaseHydrogen.run() outputs "H". Do not change or remove this line.
    releaseHydrogen.run();
    o.release();
  }

  public void oxygen(Runnable releaseOxygen) throws InterruptedException {
    o.acquire(2);
    // releaseOxygen.run() outputs "O". Do not change or remove this line.
    releaseOxygen.run();
    h.release(2);
  }
}

#include <semaphore.h>

class H2O {
private:
  sem_t h, o;
  int st;

public:
  H2O() {
    sem_init(&h, 0, 2);
    sem_init(&o, 0, 0);
    st = 0;
  }

  void hydrogen(function<void()> releaseHydrogen) {
    sem_wait(&h);
    // releaseHydrogen() outputs "H". Do not change or remove this line.
    releaseHydrogen();
    ++st;
    if (st == 2) {
      sem_post(&o);
    }
  }

  void oxygen(function<void()> releaseOxygen) {
    sem_wait(&o);
    // releaseOxygen() outputs "O". Do not change or remove this line.
    releaseOxygen();
    st = 0;
    sem_post(&h);
    sem_post(&h);
  }
};