Update main.cpp

sqrt(n) instead of n