Update main.cpp

работает. только второй критерий плохой получается на больших числах.