Update main.cpp

вот один из вариантов