Have you tried browsing through the Amiga games archive sites like
Hall Of Light or
Lemon Amiga? I can't think of any puzzle games like you discribe, but you might find the one you're looking for on one of these sites if no one else can remember the one you're thinking of!