【刷题打卡】1116. 打印零与奇偶数

266 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

一、题目描述:

1116. 打印零与奇偶数 - 力扣(LeetCode) (leetcode-cn.com)

现有函数 printNumber 可以用一个整数参数调用,并输出该整数到控制台。

  • 例如,调用 printNumber(7) 将会输出 7 到控制台。

给你类 ZeroEvenOdd 的一个实例,该类中有三个函数:zeroevenoddZeroEvenOdd 的相同实例将会传递给三个不同线程:

  • 线程 A:调用 zero() ,只输出 0
  • 线程 B:调用 even() ,只输出偶数
  • 线程 C:调用 odd() ,只输出奇数

修改给出的类,以输出序列 "010203040506..." ,其中序列的长度必须为 2n

实现 ZeroEvenOdd 类:

  • ZeroEvenOdd(int n) 用数字 n 初始化对象,表示需要输出的数。
  • void zero(printNumber) 调用 printNumber 以输出一个 0 。
  • void even(printNumber) 调用printNumber 以输出偶数。
  • void odd(printNumber) 调用 printNumber 以输出奇数。  

示例 1:

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

示例 2:

输入:n = 5
输出:"0102030405"

提示:

  • 1 <= n <= 1000

二、思路分析:

三个容量为1(容量其实无所谓,不过只有1的话比较容易理解)的队列,要注意阻塞队列不同的存取方法是有区别的。根据java官方文档:

  • 使用add方法时,如果容量不够会抛出异常;使用remove时,如果队列中没有元素会抛出异常;
  • 同样的,使用offer和poll会返回一个boolean值提示是否成功
  • 只有put和take方法会在无法put(队列已满)和无法take(队列为空)时阻塞住
  • 要注意区分不同的方法

三、AC 代码:

class ZeroEvenOdd {
    private int n;
    LinkedBlockingQueue<Integer> zero = new LinkedBlockingQueue<>(1);
    LinkedBlockingQueue<Integer> odd = new LinkedBlockingQueue<>(1);
    LinkedBlockingQueue<Integer> even = new LinkedBlockingQueue<>(1);
    
    public ZeroEvenOdd(int n) {
        this.n = n;
    }

    // printNumber.accept(x) outputs "x", where x is an integer.
    public void zero(IntConsumer printNumber) throws InterruptedException {
        zero.put(1);
        for (int i = 1; i <= n; i++) {
            try {
                zero.take();
                printNumber.accept(0);
                if (i % 2 == 1) {
                    odd.put(0);
                } else {
                    even.put(0);
                }
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    public void even(IntConsumer printNumber) throws InterruptedException {
        for (int i = 2; i <= n; i+=2) {
            try {
                even.take();
                printNumber.accept(i);
                zero.put(0);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    public void odd(IntConsumer printNumber) throws InterruptedException {
        for (int i = 1; i <= n; i+=2) {
            try {
                odd.take();
                printNumber.accept(i);
                zero.put(0);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}