задан массив из 5 целых элементов, заметить в массиве положительные элементы на единицу, а отрицательные на минус единицу​

#include <iostream>

#include <vector>  

using namespace std;

int main(){

   setlocale(LC_ALL, "Russian"); //Чтобы на русском писать текст  

   int n, i;

   cout << "Введите количество элементов в массиве : ";

   cin >> n; //Ввожу количество элементов в массиве

   vector<int>  arr(n);// объялвяю массив с n элементами

   cout << endl << "Введите элеметы массива : \n";

   //И заполнение массива,  и замена элементов на -1 и 1 будут происходить в одном цикле)

   for (i = 0; i < n; i++){

       cin >> arr[i];//Заполняю массив

       if (arr[i] > 0)//Проверяю элемент и заменяю его, но не знаю, что с нулем делать  

           arr[i] = 1;// поэтому просто оставлю его :D

       if (arr[i] < 0)

           arr[i] = -1;

   }

   cout << endl << "Массив : ";

   for (i = 0; i < n;i++)

       cout << arr[i] << "\t";

   cout << endl;

   system("pause");

   return 0;

}

a = [int(x) for x in input().split()] #вводим массив

for i in range(len(a)): # проходимся по нему выполняя условие задачи

   if a[i] > 0:

       print(1)

   elif a[i] < 0:

       print(-1)

#задача выполнена на python