1116. 打印零与奇偶数

现有函数 printNumber 可以用一个整数参数调用，并输出该整数到控制台。

例如，调用 printNumber(7) 将会输出 7 到控制台。
给你类 ZeroEvenOdd 的一个实例，该类中有三个函数：zero、even 和 odd 。ZeroEvenOdd 的相同实例将会传递给三个不同线程：

线程 A：调用 zero() ，只输出 0
线程 B：调用 even() ，只输出偶数
线程 C：调用 odd() ，只输出奇数
修改给出的类，以输出序列 “010203040506…” ，其中序列的长度必须为 2n 。

实现 ZeroEvenOdd 类：

ZeroEvenOdd(int n) 用数字 n 初始化对象，表示需要输出的数。
void zero(printNumber) 调用 printNumber 以输出一个 0 。
void even(printNumber) 调用printNumber 以输出偶数。
void odd(printNumber) 调用 printNumber 以输出奇数。

//测试代码
#include <iostream>                // std::cout
#include <thread>                // std::thread
#include <mutex>                // std::mutex, std::unique_lock
#include <condition_variable>    // std::condition_variable
#include <chrono>

std::mutex mtx; // 全局相互排斥锁.
std::condition_variable cv; // 全局条件变量.
std::mutex mtx2;
int flag = 0;
int n = 5;
using namespace std;

void printNumber(int i)
{
	//std::unique_lock <std::mutex> lck(mtx2);
	cout <<i;
}

void zero(function<void(int)> printNumber)
{
	for (size_t i = 0; i < n; i++)
	{
		std::unique_lock <std::mutex> lck(mtx);
		while (flag!=0)
		{
			cv.wait(lck);
		}
		printNumber(0);
		if (i%2==0)
		{
			flag = 1;
		}
		else
		{
			flag = 2;
		}
		cv.notify_all();
	}
}

void odd(function<void(int)> printNumber)
{
	for (size_t i = 1; i <= n; i+=2)
	{
		std::unique_lock <std::mutex> lck(mtx);
		while (flag != 1)
		{
			cv.wait(lck);
		}
		printNumber(i);
		flag = 0;
		cv.notify_all();
	}
}

void even(function<void(int)> printNumber)
{
	for (size_t i = 2; i <= n; i += 2)
	{
		std::unique_lock <std::mutex> lck(mtx);
		while (flag != 2)
		{
			cv.wait(lck);
		}
		printNumber(i);
		flag = 0;
		cv.notify_all();
	}
}

int main()
{
	n = 100;
	std::thread t1(zero, printNumber);
	std::thread t2(even, printNumber);
	std::thread t3(odd, printNumber);
	t1.join();
	t2.join();
	t3.join();
}

class ZeroEvenOdd {
private:
    int n;
    std::mutex mtx; // 全局相互排斥锁.
    std::condition_variable cv; // 全局条件变量.
    int flag=0;
public:
    ZeroEvenOdd(int n) {
        this->n = n;
    }

    // printNumber(x) outputs "x", where x is an integer.
    void zero(function<void(int)> printNumber) {
        for (size_t i = 0; i < n; i++)
        {
            std::unique_lock <std::mutex> lck(mtx);
            while (flag!=0)
            {
                cv.wait(lck);
            }
            printNumber(0);
            if (i%2==0)
            {
                flag = 1;
            }
            else
            {
                flag = 2;
            }
            cv.notify_all();
        }
    }

    void even(function<void(int)> printNumber) {
        for (size_t i = 2; i <= n; i += 2)
	{
		std::unique_lock <std::mutex> lck(mtx);
		while (flag != 2)
		{
			cv.wait(lck);
		}
		printNumber(i);
		flag = 0;
		cv.notify_all();
	}
    }

    void odd(function<void(int)> printNumber) {
        for (size_t i = 1; i <= n; i+=2)
	{
		std::unique_lock <std::mutex> lck(mtx);
		while (flag != 1)
		{
			cv.wait(lck);
		}
		printNumber(i);
		flag = 0;
		cv.notify_all();
	}
    }
};