Find the floor number with given room number?

Code

func findFloor(roomPattern: [Int], roomNo: Int) -> Int {
    var floor = 0
    var count = 0

    while count < roomNo {
        floor += 1
        let index = (floor - 1) % 2
        count += roomPattern[index]
    }

    return floor
}

Example

Test Case 1Test Case 2

// Example usage
let roomPattern = [10, 5]  // Example room pattern
let roomNo = 11         // Example room number
let result = findFloor(roomPattern: roomPattern, roomNo: roomNo)
print("Floor number: \(result)") 

Output

Floor number: 2

// Example usage
let roomPattern = [2, 3]  // Example room pattern
let roomNo = 11         // Example room number
let result = findFloor(roomPattern: roomPattern, roomNo: roomNo)
print("Floor number: \(result)") 

Output

Floor number: 5