La verdad es que tengo una duda, parece que el enunciado dice que el cohete debe empujar en la dirección de la velocidad, que no tiene por qué coincidir con el de la aceleración inicial.
Entonces no es lo que he dicho anteriormente, lo que tienes que hacer es calcular el ángulo de la velocidad para sacar los componentes x e y de la aceleración del cohete, y se los "sumas" a los ax y ay que ya tenías.
Pero como ya he dicho, mejor que preguntes a tu profesor.
Cita: lo que no sé es lo que hay que poner en el nuevo, es decir, es un problema de fisica
Esto es un foro de programación