LeetCode多线程——H2O生成(Java)
题目
现在有两种线程,氧 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。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/building-h2o
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
两种线程分别氧 oxygen 和氢 hydrogen(分别打印O和H),目标是组织这两种线程来产生水分子,即最终打印出来的序列中每3个里有2和H和一个O。
核心思想就是等待+通知,率先完成工作的A类线程需要等待B类线程也要完成工作,继而B类通知A类线程继续。我们可以使用信号量、Synchronized+wait/notify、Lock+Condition、CyclicBarrier等方法进行实现。
解法一(信号量)
生产一个H2O需要2个H和1个O,我们可以:
- 先打印2个H,再打印1个O;
- 先打印1个H,然后打印1个O,再打印一个H;
- 先打印1个O,然后再打印2和H;
以先打印2个H,再打印一个O为例,打印H的信号量hs初值为2,打印O的信号量os初值为0。
每打印一个H,需要将hs-1,同时将os+1,当我们打印完2个H时,此时由于hs=0,新的打印H的线程阻塞,os=2,可以打印一个O,同时打印一个O直接将os-2,将hs+2。此时信号量的值回到最初状态,又开始新一轮的循环。
class H2O {
private final Semaphore hs = new Semaphore(2);
private final Semaphore os = new Semaphore(0);
public H2O() {
}
public void hydrogen(Runnable releaseHydrogen) throws InterruptedException {
hs.acquire(1);
// releaseHydrogen.run() outputs "H". Do not change or remove this line.
releaseHydrogen.run();
os.release(1);
}
public void oxygen(Runnable releaseOxygen) throws InterruptedException {
os.acquire(2);
// releaseOxygen.run() outputs "O". Do not change or remove this line.
releaseOxygen.run();
hs.release(2);
}
}
解法二(信号量+内存屏障)
内存屏障(CyclicBarrier)相当于一个计数器,有两个构造方法:
- CyclicBarrier(int parties):partier表示协作线程的个数,也就是计数器的初值。简单来说就是线程调用其await()方法后(即到达这个内存屏障),线程阻塞,同时该计数器就减1,直到计数器减为0,线程从await()后面继续开始执行。
- CyclicBarrier(int parties, Runnable barrierAction):多了一个Runnable参数,当计数器减为0时,执行该barrierAction,同时阻塞线程开始执行。
基于这个两个构造函数,可以有两种思路来解决:
- 让两个线程打印H,一个线程打印O,打印完分别调用await(),这三个线程都到达CyclicBarrier,再让CyclicBarrier通知继续