1115. 交替打印FooBar

465 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动

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

解题思路

使用reentrantlock使用两个condition,实现精准唤醒

代码

    class FooBar {
        private int n;
        ReentrantLock r=new ReentrantLock();
        Condition f= r.newCondition();
        Condition b=r.newCondition();
        boolean flag=true;
        public FooBar(int n) {
            this.n = n;
        }

        public void foo(Runnable printFoo) throws InterruptedException {
            r.lock();
            for (int i = 0; i < n; i++) {
                while (!flag)
                    f.await();
                // printFoo.run() outputs "foo". Do not change or remove this line.
                printFoo.run();
                flag=false;
                b.signal();
            }
            r.unlock();
        }

        public void bar(Runnable printBar) throws InterruptedException {
            r.lock();
            for (int i = 0; i < n; i++) {
                while (flag)
                    b.await();
                // printBar.run() outputs "bar". Do not change or remove this line.
                printBar.run();
                flag=true;
                f.signal();
            }
            r.unlock();
        }
    }

解题思路

使用synchronized,加Boolean类型的变量,通过互斥以及标志变量,实现交替打印

class FooBar {
    private int n;
    private boolean forb=true;
    public FooBar(int n) {
        this.n = n;
    }

    public void foo(Runnable printFoo) throws InterruptedException {

        for (int i = 0; i < n; i++) {
            synchronized(this)
            {
                while(!forb)
                {wait();}
                printFoo.run();
                forb=false;
                notifyAll();
            }
            // printFoo.run() outputs "foo". Do not change or remove this line.

        }
    }

    public void bar(Runnable printBar) throws InterruptedException {

        for (int i = 0; i < n; i++) {
            synchronized(this)
            {
                while(forb)
                {wait();}
                printBar.run();
                forb=true;
                notifyAll();

            }
            // printBar.run() outputs "bar". Do not change or remove this line.

        }
    }
}