Вставка элемента

[Dmitry73]

необходимо вставить после элементов массива являющихся полными квадратами их корни.
У меня если корень тоже является полным квадратом его тоже раскладывает на квадрат, и на еденице вообще не работает. Пробовал исправить по разному, неполучается. помогите пожалуйста.

#define N 100
using namespace std;
int koren(int a[],int i)
{int j=0,flag=0,b=0,k=0;
for(j=1;j<a[i];j++)
{if(j*j==a[i])
{k=j;
flag=1;
b=k;}
}
if(flag==1)
return b;
else
return 0;
}




int main()
{int a[N],i,j,n=10;
cout<<"vvedite massiv \n";
for(i=0;i<n;i++)
{cin>>a[i];}
for(i=0;i<n;i++)
{
if(koren(a,i)!=0)
{n++;
for(j=n-1;j>i;j--)
{a[j+1]=a[j];
}
a[i+1]=koren(a,i);
}
}
for(i=0;i<n;i++)
cout<<a[i]<<" ";

}