02 Hospital -For Loop More Exercises

function solve(...input) {
    let days = Number(input.shift());
    let treatedPeople = 0;
    let untreatedPeople = 0;
    let doctors = 7;

    for(let i = 1; i <= days; i++) {
        let patients = Number(input.shift());

        if(i % 3 == 0 && untreatedPeople > treatedPeople) {
            doctors++;
        }

        if (patients <= doctors) {
            treatedPeople += patients;
        } else {
            untreatedPeople += patients - doctors;
            treatedPeople += doctors;
        }

    }

    console.log(`Treated patients: ${treatedPeople}.`);
    console.log(`Untreated patients: ${untreatedPeople}.`);
}