Yes, there is no need to prepare separately. In Mains, the problem would be based on one concept and not involve much mathematical work. In the JEE Advance exam, you will be presented with a combination of problems. But then the concepts would remain the same.
Picture Credit : Google