Массивы. Вставить новый элемент после всех элементов, которые заканчиваются на данную цифру

#include <iostream>
#include <ctime>
#include <cstdlib>
 
void fill_arr(int * arr, int size)
{
    for(int i = 0; i < size; ++i)
    {
        arr[i] = rand() % 100;
    }
}
 
void output_arr(int * arr, int size)
{
    for(int i = 0; i < size; ++i)
    {
        std::cout << arr[i] << " ";
    }
}
 
int count_last(int * arr, int size, int digit)
{
    int count = 0;
    
    for(int i = 0; i < size; ++i)
    {
        if(arr[i] % 10 == digit)
            ++count;
    }
    
    return count;
}
 
void insert_elems(int *& arr, int & size, int elem, int digit)
{
    int count  = count_last(arr, size, digit);
    
    if(!count)
        return;
        
    std::cout << count << "\n";
    int * temp = new int[size + count];
    
    for(int i = 0, j = 0; i < size; ++i, ++j)
    {
        temp[j] = arr[i];
        
        if(arr[i] % 10 == digit)
        {
            temp[++j] = elem;
        }
    }
    
    delete [] arr;
    
    arr = temp;
    
    size += count;
}
        
 
int main()
{
    srand(static_cast<unsigned int>(time(0))); 
    
    int size, elem, digit;
 
    std::cout << "Enter size: ";
    std::cin >> size;
 
    int * arr = new int [size];
    
    
    fill_arr(arr, size);
    std::cout << "Source:\n";
    output_arr(arr, size);
    
    std::cout << "\n\nEnter new element: ";
    std::cin >> elem;
    std::cout << "\n\nEnter last digit: ";
    std::cin >> digit;
    
    insert_elems(arr, size, elem, digit);
    std::cout << "\nResult:\n";
    output_arr(arr, size);
    
   
    delete [] arr;
   
    return 0;
}