I think looking at the resulting graph of Neural Algorithmic Learner (Petr Velickovic and the group at Deepmind) may result in an algorithm that won't be able to generate code directly.

Bonus, there is this CLRS benchmark that might be useful for the evaluation of your task.