Написать функцию, которая получает указатель на динамический массив и его размер. Функция должна удалить из массива все отрицательные числа и вернуть указатель на новый динамический массив.

#include <iostream>

int* removeNegativeNumbers(int* arr, int size) { int count = 0;

// Подсчитываем количество неотрицательных чисел
for (int i = 0; i < size; i++) {
    if (arr[i] >= 0) {
        count++;
    }
}

// Создаем новый динамический массив
int* newArr = new int[count];

// Копируем неотрицательные числа в новый массив
int j = 0;
for (int i = 0; i < size; i++) {
    if (arr[i] >= 0) {
        newArr[j] = arr[i];
        j++;
    }
}

return newArr;

}

int main() { int size; std::cout << "Enter the size of the array: "; std::cin >> size;

int* arr = new int[size];

std::cout << "Enter the elements of the array: ";
for (int i = 0; i < size; i++) {
    std::cin >> arr[i];
}

int* newArr = removeNegativeNumbers(arr, size);

std::cout << "New array without negative numbers: ";
for (int i = 0; i < size; i++) {
    std::cout << newArr[i] << " ";
}

delete[] arr;
delete[] newArr;

return 0;

}