Yep, I once wrote something like this in Prolog as an assignment. I don't remember the details and the code is long lost though. Anyway if I were to do it again (and had to implement it myself) I'd definitely use Prolog again. It does take some time to get used to it since you really have to think differently than when using imperative languages, but once you get the twist it's great for this kind of tasks.