On the way back there's nothing that stops you buying a domestic flight. Indian domestic flights and trains cost the same whether booked 2 x one way or 1 x roundtrip.