leetcode 1117. H2O 生成（C++）

现在有两种线程，氧 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 。

C++

class H2O 
{
private:
	std::condition_variable cv;
	vector<int> tmp{0,0,0};
	mutex Mu;

public:
    H2O() 
    {

    }

    void hydrogen(function<void()> releaseHydrogen) 
    {
        std::unique_lock<mutex> lk(Mu);
        cv.wait(lk,[this](){return tmp[0]==0 || tmp[1]==0;});
        // releaseHydrogen() outputs "H". Do not change or remove this line.
        releaseHydrogen();
        if(0==tmp[0])
        {
        	tmp[0]=1;
        }
        else
        {
        	tmp[1]=1;
        }
        if(1==tmp[0] && 1==tmp[1] && 1==tmp[2])
        {
        	tmp={0,0,0};
        }
        lk.unlock();
        cv.notify_all();
    }

    void oxygen(function<void()> releaseOxygen) 
    {
        std::unique_lock<mutex> lk(Mu);
        cv.wait(lk,[this](){return tmp[2]==0;});
        // releaseOxygen() outputs "O". Do not change or remove this line.
        releaseOxygen();
        tmp[2]=1;
        if(1==tmp[0] && 1==tmp[1] && 1==tmp[2])
        {
        	tmp={0,0,0};
        }
        lk.unlock();
        cv.notify_all();
    }
};