Имеется массив k из n чисел. создайте алгоритм решения замены всех элементов массива, меньших 5, на квадраты этих элементов. если таких элементов нет, то выдать соответствующее сообщение.

//c++
#include <bits/stdc++.h>

using namespace std;

int main()
{
     long long n;
     bool z=0;
     cin>>n;
     long long k[n];
     for (long long i=0; i<n; i++)
    {
         cin>>k[i];
         if(k[i]<5) {k[i]=k[i]*k[i]; z=1;} 
    }
    if (z==0) cout<<"таких элементов нет"; else {
    for (long long i=0; i<n; i++)
            cout<< k[i]<<" ";
                                                                            }
return 0;
}