function solve(arr) {


  let energy = Number(arr.shift());

  let i = 0;

  let counter = 0;


  while (arr[i] !== 'End of battle') {

    let distance = Number(arr[i]);

    if (energy - distance >= 0) {

    energy -= distance;

    counter++;

    if (counter % 3 === 0) {

      energy += counter;

    }

  } else {

    console.log(`Not enough energy! Game ends with ${counter} won battles and ${energy} energy`);

    return;

  }

  i++;

  }




    console.log(`Won battles: ${counter}. Energy left: ${energy}`);

  }