题目描述:
现在有两种线程,氧 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 。
方法1:使用两个信号量实现
主要思路:
(1)使用两个信号量分别标识两个生产两种原子的线程,将标识生产h的线程的信号量初始化为 2,将标识生产o 的信号量初始化为 0,让两个信号量配合实现交替输出;
(2)该方法的缺点是,只能输出固定顺序的h和o的组成顺序,既只能输出HHO组合;
(3)使用互斥量和条件变量类似的缺点;
#include<semaphore.h>
class H2O {
public:
sem_t sem_h;
sem_t sem_o;
int count_h=0;
H2O() {
sem_init(&sem_h,0,2);
sem_init(&sem_o,0,0);
}
void hydrogen(function<void()> releaseHydrogen) {
sem_wait(&sem_h);
++count_h;
// releaseHydrogen() outputs "H". Do not change or remove this line.
releaseHydrogen();
if(count_h==2){//等于2时,说明可以释放o了,故对sem_o调用post,同时将 count_h=0;进行下一轮
count_h=0;
sem_post(&sem_o);
}
}
void oxygen(function<void()> releaseOxygen) {
sem_wait(&sem_o);
// releaseOxygen() outputs "O". Do not change or remove this line.
releaseOxygen();
sem_post(&sem_h);
sem_post(&sem_h);
}
};
方法2:使用条件变量和互斥量实现,但超时了
主要思路:
(1)和方法1思路类似;
#include<pthread.h>
class H2O {
public:
pthread_mutex_t mutex;
pthread_cond_t cond_h;
pthread_cond_t cond_o;
int count_h;
H2O() {
count_h=0;
pthread_cond_init(&cond_h,0);
pthread_cond_init(&cond_o,0);
pthread_mutex_init(&mutex,0);
}
void hydrogen(function<void()> releaseHydrogen) {
pthread_mutex_lock(&mutex);
while(count_h==2){
pthread_cond_wait(&cond_h,&mutex);
}
// releaseHydrogen() outputs "H". Do not change or remove this line.
releaseHydrogen();
++count_h;
pthread_cond_signal(&cond_o);
}
void oxygen(function<void()> releaseOxygen) {
pthread_mutex_lock(&mutex);
while(count_h<2){
pthread_cond_wait(&cond_o,&mutex);
}
// releaseOxygen() outputs "O". Do not change or remove this line.
releaseOxygen();
count_h=0;
pthread_cond_signal(&cond_h);
}
};