function solve(input){
    let initialEnergy = input.shift();
    let counter = 0;
    let commands = {
        end: "End of battle"
    }
    for(let i = 0; i < input.length; i++) {
        if(input[i] === commands.end) {
            console.log(`Won battles: ${counter}. Energy left: ${initialEnergy}`)
            break;
        }
        if(initialEnergy >= parseInt(input[i])) {
            initialEnergy = initialEnergy - parseInt(input[i]);
            counter++;
            if(counter % 3 == 0) {
                initialEnergy = initialEnergy + counter;
            }
        } else {
            console.log(`Not enough energy! Game ends with ${counter} won battles and ${initialEnergy} energy`)
            break;
        }
    }
}