You can do whatever you want - it just has to return the correct result.

It isn't that difficult to solve, but I like the idea.