欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode 1116 打印零和奇偶数 附java代码

程序员文章站 2022-04-02 18:06:30
...

1. 题目要求

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

题目已经很清楚了,这涉及到三个线程之间的通信。

这里我们用synchronized来实现

LeetCode 1116 打印零和奇偶数 附java代码

 

class ZeroEvenOdd {
    private int n;
    
    private  AtomicInteger start = new AtomicInteger(1);

    private  AtomicBoolean flag = new AtomicBoolean(true);

    public ZeroEvenOdd(int n) {
        this.n = n;
    }

    // printNumber.accept(x) outputs "x", where x is an integer.
    public void zero(IntConsumer printNumber) throws InterruptedException {
        
         while (start.get() <= n) {

            synchronized (start) {
                if (flag.get()) {
                    printNumber.accept(0);
                    flag.set(false);
                    start.notifyAll();
                } else {
                    start.wait();
                }
            }

        }

    }

    public void even(IntConsumer printNumber) throws InterruptedException {
        
         while (start.get() <= n) {
            synchronized (start) {
                if (!flag.get() && start.get() % 2 == 0) {
                    printNumber.accept(start.get());
                    start.getAndIncrement();
                    flag.set(true);
                    start.notifyAll();
                } else {
                    start.wait();
                }
            }
        }

    }

    public void odd(IntConsumer printNumber) throws InterruptedException {
        while (start.get() <= n) {
            synchronized (start) {
                if (!flag.get() && start.get() % 2 != 0) {
                    printNumber.accept(start.get());
                    start.getAndIncrement();
                    flag.set(true);
                    start.notifyAll();
                } else {
                    start.wait();
                }
            }
        }

    }
}