本文已参与「新人创作礼」活动,一起开启掘金创作之路。
一、题目描述:
1116. 打印零与奇偶数 - 力扣(LeetCode) (leetcode-cn.com)
现有函数 printNumber 可以用一个整数参数调用,并输出该整数到控制台。
- 例如,调用
printNumber(7)将会输出7到控制台。
给你类 ZeroEvenOdd 的一个实例,该类中有三个函数:zero、even 和 odd 。ZeroEvenOdd 的相同实例将会传递给三个不同线程:
- 线程 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();
}
}
}
}