1115 - 交替打印FooBar

题目描述

我们提供一个类：

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" 将被输出两次。

链接：https://leetcode-cn.com/problems/print-foobar-alternately

题解

class FooBar {
    private int n;
    private Lock lock;
    private volatile boolean flag;

    public FooBar(int n) {
        this.n = n;
        this.lock = new ReentrantLock(true);
        this.flag = true;
    }

    public void foo(Runnable printFoo) throws InterruptedException {
        
        for (int i = 0; i < n;) {
            lock.lock();
            try {
                if(flag) {
                    // printFoo.run() outputs "foo". Do not change or remove this line.
                    printFoo.run();
                    i++;
                    flag = false;
                }
            } finally {
                lock.unlock();
            }
        }
    }

    public void bar(Runnable printBar) throws InterruptedException {
        
        for (int i = 0; i < n;) {
            lock.lock();
            try {
                if(!flag) {
                    // printBar.run() outputs "bar". Do not change or remove this line.
        	        printBar.run();
                    i++;
                    flag = true;
                }
            } finally {
                lock.unlock();
            }
        }
    }
}

返回