多线程-打印零和奇偶数

打印零和奇偶数

要求

假设有这么一个类：

class ZeroEvenOdd {
  public ZeroEvenOdd(int n) { ... }      // 构造函数
  public void zero(printNumber) { ... }  // 仅打印出 0
  public void even(printNumber) { ... }  // 仅打印出 偶数
  public void odd(printNumber) { ... }   // 仅打印出 奇数
}

相同的一个 ZeroEvenOdd 类实例将会传递给三个不同的线程：

1.线程 A 将调用 zero()，它只输出 0 。
2.线程 B 将调用 even()，它只输出偶数。
3.线程 C 将调用 odd()，它只输出奇数。
每个线程都有一个 printNumber 方法来输出一个整数。请修改给出的代码以输出整数序列 010203040506… ，其中序列的长度必须为 2n。

示例 1：

输入：n = 2
输出：“0102”
说明：三条线程异步执行，其中一个调用 zero()，另一个线程调用 even()，最后一个线程调用odd()。正确的输出为 “0102”。

示例 2：

输入：n = 5
输出：“0102030405”

该例子的测试代码如下：

#include <iostream>
#include <functional>
#include <semaphore.h>
#include <thread>

using namespace std; 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */

class ZeroEvenOdd {
private:
    int n;
	pthread_mutex_t zero_num = PTHREAD_MUTEX_INITIALIZER;
	pthread_mutex_t nonzero_num1 = PTHREAD_MUTEX_INITIALIZER;
	pthread_mutex_t nonzero_num2 = PTHREAD_MUTEX_INITIALIZER;
	//核心思想：串行输出变成并行输出
	//即先抽取成两个流，非零流和0流
	//然后将非零流抽取成奇数流和偶数流
	//这样的两个流可以采用互斥量加锁的方式 
public:
    ZeroEvenOdd(int n) {
        this->n = n;
        pthread_mutex_lock(&nonzero_num1);
        pthread_mutex_lock(&nonzero_num2);
    }

    // printNumber(x) outputs "x", where x is an integer.
    //零 
    void zero(function<void(int)> printNumber) {
        for(int i=0;i<n;i++){
        	pthread_mutex_lock(&zero_num);
        	printNumber(0);
        	if(i&1==1){
        		pthread_mutex_unlock(&nonzero_num2);
			}else{
				pthread_mutex_unlock(&nonzero_num1);
			} 
		}
    }
	//偶数 
    void even(function<void(int)> printNumber) {
        for(int i=2;i<=n;i=i+2){
        	pthread_mutex_lock(&nonzero_num2);
        	printNumber(i);
        	pthread_mutex_unlock(&zero_num);
		} 
    }
	//奇数 
    void odd(function<void(int)> printNumber) {
        for(int i=1;i<=n;i=i+2){
        	pthread_mutex_lock(&nonzero_num1);
        	printNumber(i);
        	pthread_mutex_unlock(&zero_num);
		}
    }
};

static ZeroEvenOdd *obj;

class Print{
public:
	static void print(int a){
		cout<<a;
	}
	
	static void fun1(){
		obj->zero(print);
	}
	
	static void fun2(){
		obj->even(print);	
	}
	
	static void fun3(){
		obj->odd(print);
	}
};

int main(int argc, char** argv) {
	obj = new ZeroEvenOdd(10);
	thread t1(Print::fun1);
	thread t2(Print::fun2);
	thread t3(Print::fun3);
	
	t1.join();
	t2.join();
	t3.join();

	return 0; 
}