[力扣]1114. 按序打印

按序打印

Posted by JinFei on April 12, 2020

题目描述

public class Foo {   public void one() { print(“one”); }   public void two() { print(“two”); }   public void three() { print(“three”); } }

三个不同的线程将会共用一个 Foo 实例。
线程 A 将会调用 one() 方法
线程 B 将会调用 two() 方法
线程 C 将会调用 three() 方法
请设计修改程序,以确保 two() 方法在 one() 方法之后被执行,three() 方法在 two() 方法之后被执行。

Example1:

  输入: [1,2,3]
    输出: "onetwothree"
    解释: 
    有三个线程会被异步启动。
    输入 [1,2,3] 表示线程 A 将会调用 one() 方法,线程 B 将会调用 two() 方法,线程 C 将会调用 three() 方法。
    正确的输出是 "onetwothree"。

Example2:

  输入: [1,3,2]
    输出: "onetwothree"
    解释: 
    输入 [1,3,2] 表示线程 A 将会调用 one() 方法,线程 B 将会调用 three() 方法,线程 C 将会调用 two() 方法。
    正确的输出是 "onetwothree"。

注意

  • 尽管输入中的数字似乎暗示了顺序,但是我们并不保证线程在操作系统中的调度顺序。
  • 你看到的输入格式主要是为了确保测试的全面性。

思路

  • 使用条件变量和互斥锁
class Foo {
private:
    condition_variable data_vari;
    mutex data_mutex;
    int num;
public:
    Foo() {
        num = 1;
    }

    void first(function<void()> printFirst) {
        unique_lock<mutex> ulock(data_mutex);
        // printFirst() outputs "first". Do not change or remove this line.
        printFirst();
        ++num;
        data_vari.notify_all();
    }

    void second(function<void()> printSecond) {
        unique_lock<mutex> ulock(data_mutex);
        data_vari.wait(ulock, [this](){return num == 2;});
        // printSecond() outputs "second". Do not change or remove this line.
        printSecond();
        ++num;
        data_vari.notify_all();
    }

    void third(function<void()> printThird) {
        unique_lock<mutex> ulock(data_mutex);
        data_vari.wait(ulock, [this](){return num == 3;});
        // printThird() outputs "third". Do not change or remove this line.
        printThird();
        data_vari.notify_all();
    }
};