Medium
Print Zero Even Odd — C++
Full explanation · Time O(n) · Space O(1)
// Time: O(n)
// Space: O(1)
class ZeroEvenOdd {
public:
ZeroEvenOdd(int n) : n_(n) {
}
// printNumber(x) outputs "x", where x is an integer.
void zero(function<void(int)> printNumber) {
for (int i = 0; i < n_; ++i) {
{
unique_lock<mutex> l(m_);
wait_.wait(l, [this]() { return curr_ % 2 == 0; });
++curr_;
printNumber(0);
}
wait_.notify_all();
}
}
void even(function<void(int)> printNumber) {
for (int i = 2; i <= n_; i += 2) {
{
unique_lock<mutex> l(m_);
wait_.wait(l, [this]() { return curr_ % 4 == 3; });
++curr_;
printNumber(i);
}
wait_.notify_all();
}
}
void odd(function<void(int)> printNumber) {
for (int i = 1; i <= n_; i += 2) {
{
unique_lock<mutex> l(m_);
wait_.wait(l, [this]() { return curr_ % 4 == 1; });
++curr_;
printNumber(i);
}
wait_.notify_all();
}
}
private:
int n_;
int curr_ = 0;
mutex m_;
condition_variable wait_;
};
// Time: O(n)
// Space: O(1)
class ZeroEvenOdd2 {
public:
ZeroEvenOdd2(int n) : n_(n) {
m2_.lock();
m3_.lock();
}
// printNumber(x) outputs "x", where x is an integer.
void zero(function<void(int)> printNumber) {
for (int i = 0; i < n_; ++i) {
m1_.lock();
printNumber(0);
if (++curr_ % 2) {
m2_.unlock();
} else {
m3_.unlock();
}
}
}
void even(function<void(int)> printNumber) {
for (int i = 2; i <= n_; i += 2) {
m3_.lock();
printNumber(curr_);
m1_.unlock();
}
}
void odd(function<void(int)> printNumber) {
for (int i = 1; i <= n_; i += 2) {
m2_.lock();
printNumber(curr_);
m1_.unlock();
}
}
private:
int n_;
int curr_ = 0;
mutex m1_, m2_, m3_;
};