function solve(capacity, wagons) {
    let difference = 0;
    let train = [];
    for (let i = 0; i < wagons.length; i++) {
        if (wagons[i] <= capacity) {
            train[i] = wagons[i];
        } if (wagons[i] > capacity && difference === 0) {
           
            train[i] = capacity;
        } if (wagons[i] + difference <= capacity) {
            train[i] = wagons[i] + difference;
            difference = 0;
        } if (wagons[i] + difference > capacity) {
            difference = (wagons[i] + difference) - capacity;
            train[i] = capacity;
        }

    }
    if (difference === 0) {
        console.log(train);
        console.log('All passengers aboard');
    } else {
        console.log(train);
        console.log(`Could not fit ${difference} passengers`);

    }

}