1117. H2O 生成
现在有两种线程,氧 oxygen 和氢 hydrogen,你的目标是组织这两种线程来产生水分子。
存在一个屏障(barrier)使得每个线程必须等候直到一个完整水分子能够被产生出来。
氢和氧线程会被分别给予 releaseHydrogen 和 releaseOxygen 方法来允许它们突破屏障。
这些线程应该三三成组突破屏障并能立即组合产生一个水分子。
你必须保证产生一个水分子所需线程的结合必须发生在下一个水分子产生之前。
换句话说:
如果一个氧线程到达屏障时没有氢线程到达,它必须等候直到两个氢线程到达。
如果一个氢线程到达屏障时没有其它线程到达,它必须等候直到一个氧线程和另一个氢线程到达。
书写满足这些限制条件的氢、氧线程同步代码。
- 示例 1:
输入: “HOH”
输出: “HHO”
解释: “HOH” 和 “OHH” 依然都是有效解。
- 示例 2:
输入: “OOHHHH”
输出: “HHOHHO”
解释: “HOHHHO”, “OHHHHO”, “HHOHOH”, “HOHHOH”, “OHHHOH”, “HHOOHH”, “HOHOHH” 和 “OHHOHH” 依然都是有效解。
提示:
输入字符串的总长将会是 3n, 1 ≤ n ≤ 50;
输入字符串中的 “H” 总数将会是 2n 。
输入字符串中的 “O” 总数将会是 n 。
解题思路
- 当h已经产生两个而且还没消费的情况下,我们要阻塞产生h的线程
- 当0已经产生两个而且还没消费的情况下,我们要阻塞产生0的线程
- 如果当前线程生产完以后,刚好可以产生hho,我们就将这h和o的计数清零,并且唤醒之前被阻塞的线程
代码
class H2O {public H2O() {}ReentrantLock reentrantLock=new ReentrantLock();Condition condition=reentrantLock.newCondition();volatile int h=0,o=0;public void hydrogen(Runnable releaseHydrogen) throws InterruptedException {reentrantLock.lock(); while(h==2)condition.await();h++;if (h==2&&o==1){h=0;o=0;}releaseHydrogen.run();condition.signalAll();reentrantLock.unlock();// releaseHydrogen.run() outputs "H". Do not change or remove this line.}public void oxygen(Runnable releaseOxygen) throws InterruptedException {reentrantLock.lock(); while(o==1)condition.await();o++;if (h==2&&o==1){h=0;o=0;}// releaseOxygen.run() outputs "O". Do not change or remove this line.releaseOxygen.run(); condition.signalAll();reentrantLock.unlock();}}