I wrote it that way, but for some reason I got an error.

Every time I write a solution to a problem, I think I've come up with a very cool solution. But then I look at other people's solutions and see that they are much more elegant and simple than mine. Tell me, am I learning really badly, or do I just not have enough knowledge yet and over time I will learn to write as elegantly as the rest?

Oh, I made such a construction for a function, but it turns out there are so many very simple solutions in one line...

This is probably the most terrible task I've ever seen, especially when you're solving it on swift and you're a beginner...