交替打印FooBar （ 多线程 / LeetCode ）

题目链接

题目描述

我们提供一个类：

class FooBar {
  public void foo() {
    for (int i = 0; i < n; i++) {
      print("foo");
    }
  }

  public void bar() {
    for (int i = 0; i < n; i++) {
      print("bar");
    }
  }
}

两个不同的线程将会共用一个 FooBar 实例。其中一个线程将会调用 foo() 方法，另一个线程将会调用 bar() 方法。

请设计修改程序，以确保 “foobar” 被输出 n 次。

示例 1:

输入: n = 1
输出: "foobar"
解释: 这里有两个线程被异步启动。其中一个调用 foo() 方法, 另一个调用 bar() 方法，"foobar" 将被输出一次。

示例 2:

输入: n = 2
输出: "foobarfoobar"
解释: "foobar" 将被输出两次。

思路

使用互斥锁和条件变量解决，通过flag的值来决定执行顺序，保证foo和bar交替输出。

代码如下：

class FooBar {
private:
    int n;
    int flag;
    pthread_cond_t cond;
    pthread_mutex_t mutex;
public:
    FooBar(int n) {
        this->n = n;
        flag=0;
        pthread_mutex_init(&mutex, NULL);
        pthread_cond_init(&cond, NULL);
    }

    void foo(function<void()> printFoo) {    
        for (int i = 0; i < n; i++) { 
        	pthread_mutex_lock(&mutex);
            while(flag!=0){
               pthread_cond_wait(&cond,&mutex);
            }
            printFoo();
            flag=1;
            pthread_cond_broadcast(&cond);
            pthread_mutex_unlock(&mutex);
        }
    }

    void bar(function<void()> printBar) {
        for (int i = 0; i < n; i++) {
            pthread_mutex_lock(&mutex);
            while(flag==0){
               pthread_cond_wait(&cond,&mutex);
            }
        	printBar();
            flag=0;
            pthread_cond_broadcast(&cond);
            pthread_mutex_unlock(&mutex);
        }
        
    }
};