You could check out the following books, in no particular order, Puzzles to Puzzle You by Shakuntata Devi, Mind Benders Brain Teasers by Vikas Khatri, Puzzles & Teasers by George J. Summers. With that being said, I would recommend you to practice puzzles from geeksforgeeks, instead of any books because geeksforgeeks has a list of questions that have been asked previously to other students plus you get answers to them easily on the internet. You can also try brainstellar, but that is more focused to probability types puzzles asked in Finance (and big tech) industry.