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

1. 题目要求

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

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

这里我们用synchronized来实现

 

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();
                }
            }
        }

    }
}