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();
    }
};