Solved with Z-function in 17 rows of code (if don't count 10 rows for Z-function)