Ты язык не перепутал? Такое легко провернуть на Python, JavaScript, PHP. Более того, сама эта задача взята из задачника Python.
Но пытаться сделать подобное на C++ - совершеннейшее извращение. И дело даже не в разных типах параметра, а в разных типах возвращаемого значения.