Заполнить массив случайными числами так чтобы обязательно был хотя бы один ноль - C++ - Ответ 13608410

#include <cstddef>
#include <iostream>
#include <random>
 
int main()
{
    std::cout << "Enter length of Array: ";
    std::size_t size;
    std::cin >> size;
 
    int* p = new int[size];
 
    if (size == 0)
    {
        std::cerr << "Error";
        return -1;
    }
 
    if (size == 1)
    {
        *p = 0;
        std::cout << "Your Array:" << std::endl;
        std::cout << *p;
 
        delete[] p;
        return 0;
    }
 
    std::default_random_engine rand(std::random_device{}() );
    std::uniform_int_distribution<> dist{0, 1};
 
    for (std::size_t offset = 0; offset < size; ++offset)
        *(p + offset) = dist(rand);
 
    std::cout << "Your Array:" << std::endl;
    for (std::size_t offset = 0; offset < size; ++offset)
        std::cout << *(p + offset) << ' ';
 
    delete[] p;
    return 0;
}