When as expression returning value in Kotlin - Time & Space Complexity

Time Complexity: When as expression returning value

O(n)

Understanding Time Complexity

We want to understand how the time taken by a Kotlin when expression changes as the input changes.

Specifically, how does the number of checks grow when using when to return a value?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

fun getDayType(day: Int): String = when(day) {
    1 -> "Monday"
    2 -> "Tuesday"
    3 -> "Wednesday"
    4 -> "Thursday"
    5 -> "Friday"
    6 -> "Saturday"
    7 -> "Sunday"
    else -> "Invalid day"
}
    

This code returns the name of the day for a given number using when as an expression.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Checking each case in the when expression.
• How many times: Up to 7 checks in the worst case before matching or hitting else.

How Execution Grows With Input

As the input number changes, the when expression checks cases one by one until it finds a match.

Input Size (n)	Approx. Operations
3	3 checks
5	5 checks
7	7 checks

Pattern observation: The number of checks grows linearly with the number of cases.

Final Time Complexity

Time Complexity: O(n)

This means the time to find the matching case grows in a straight line as the number of cases increases.

Common Mistake

[X] Wrong: "The when expression always runs in constant time no matter how many cases it has."

[OK] Correct: Actually, the when expression checks cases one by one until it finds a match, so more cases mean more checks and longer time.

Interview Connect

Understanding how when expressions scale helps you explain how your code behaves with more conditions, showing you think about efficiency clearly.

Self-Check

"What if we replaced the when expression with a map lookup? How would the time complexity change?"

Practice

1. What does the when expression in Kotlin return when used as an expression?

easy

A. It returns a value based on the matched case.

B. It always returns Unit (no value).

C. It returns a Boolean indicating if a case matched.

D. It throws an exception if no case matches.

Solution

1. Step 1: Understand when as expression

   In Kotlin, when can be used as an expression that returns a value depending on which case matches.

2. Step 2: Compare with other options

   It does not always return Unit, nor a Boolean, and it requires an else branch to avoid exceptions, so it won't throw by default.

3. Final Answer:

   It returns a value based on the matched case. -> Option A

4. Quick Check:

   when returns value [OK]

Hint: Remember: when returns the matched case value [OK]

Common Mistakes:

• Thinking when always returns Unit
• Confusing when with if returning Boolean
• Forgetting to add an else branch

2. Which of the following is the correct syntax to use when as an expression returning a value in Kotlin?

easy

A. val result = when x { 1 -> "One"; else -> "Other" }

B. val result = when (x) { x > 0 -> "Positive" else -> "Non-positive" }

C. val result = when { x > 0 -> "Positive"; else -> "Non-positive" }

D. val result = when (x) { in 1..10 -> "Small"; else -> "Large" }

Solution

1. Step 1: Check correct when syntax with argument

   val result = when (x) { in 1..10 -> "Small"; else -> "Large" } uses when (x) with cases separated by semicolons and an else branch, which is correct.

2. Step 2: Identify syntax errors in other options

   val result = when { x > 0 -> "Positive"; else -> "Non-positive" } uses semicolons inside braces correctly; val result = when (x) { x > 0 -> "Positive" else -> "Non-positive" } misses semicolon before else; val result = when x { 1 -> "One"; else -> "Other" } misses parentheses around x.

3. Final Answer:

   val result = when (x) { in 1..10 -> "Small"; else -> "Large" } -> Option D

4. Quick Check:

   Correct when syntax uses parentheses and semicolons [OK]

Hint: Use parentheses around variable and semicolons between cases [OK]

Common Mistakes:

• Omitting parentheses around the variable
• Using commas instead of semicolons
• Missing semicolon before else branch

3. What is the output of the following Kotlin code?

val x = 5
val result = when {
  x < 0 -> "Negative"
  x == 0 -> "Zero"
  x in 1..10 -> "Small"
  else -> "Large"
}
println(result)

medium

A. Small

B. Zero

C. Negative

D. Large

Solution

1. Step 1: Evaluate conditions in order

   x = 5 is not less than 0, not equal to 0, but is in 1..10 range, so the third case matches.

2. Step 2: Determine returned value

   The matched case returns "Small", so result is "Small".

3. Final Answer:

   Small -> Option A

4. Quick Check:

   5 in 1..10 = true, returns "Small" [OK]

Hint: Check conditions top to bottom; first true case returns value [OK]

Common Mistakes:

• Assuming else branch runs for 5
• Confusing equality and range checks
• Ignoring order of conditions

4. Identify the error in this Kotlin code using when as an expression:

val x = 3
val result = when (x) {
  1 -> "One"
  2 -> "Two"
}
println(result)

medium

A. Incorrect variable name in when.

B. Missing else branch causes compilation error.

C. Semicolons missing between cases.

D. Cannot use when as expression.

Solution

1. Step 1: Check completeness of when expression

   The when expression covers only 1 and 2, but x = 3 is not handled and no else branch is provided.

2. Step 2: Understand Kotlin requirement for exhaustive when

   When used as an expression, when must be exhaustive, so missing else causes a compilation error.

3. Final Answer:

   Missing else branch causes compilation error. -> Option B

4. Quick Check:

   Exhaustive when needs else [OK]

Hint: Always add else if not all cases covered [OK]

Common Mistakes:

• Ignoring missing else branch
• Thinking semicolons are mandatory
• Assuming when always compiles without else

5. You want to assign a grade based on a score using when as an expression. Which code correctly returns "A" for scores 90 and above, "B" for 80-89, "C" for 70-79, and "F" otherwise?

hard

A. val grade = when (score) {
score >= 90 -> "A"
score >= 80 -> "B"
score >= 70 -> "C"
else -> "F"
}

B. val grade = when (score) {
in 90..100 -> "A"
in 80..89 -> "B"
in 70..79 -> "C"
else -> "F"
}

C. val grade = when {
score >= 90 -> "A"
score >= 80 -> "B"
score >= 70 -> "C"
else -> "F"
}

D. val grade = when {
in 90..100 -> "A"
in 80..89 -> "B"
in 70..79 -> "C"
else -> "F"
}

Solution

1. Step 1: Understand when with condition expressions

   val grade = when { score >= 90 -> "A" score >= 80 -> "B" score >= 70 -> "C" else -> "F" } uses when without argument and conditions like score >= 90, which is valid and clear for ranges.

2. Step 2: Check other options for errors

   val grade = when (score) { in 90..100 -> "A" in 80..89 -> "B" in 70..79 -> "C" else -> "F" } is correct syntax but uses ranges; val grade = when (score) { score >= 90 -> "A" score >= 80 -> "B" score >= 70 -> "C" else -> "F" } incorrectly uses conditions inside when(score) which expects values, not conditions; val grade = when { in 90..100 -> "A" in 80..89 -> "B" in 70..79 -> "C" else -> "F" } uses in without variable, which is invalid.

3. Final Answer:

   val grade = when { score >= 90 -> "A" score >= 80 -> "B" score >= 70 -> "C" else -> "F" } -> Option C

4. Quick Check:

   Use when without argument for conditions [OK]

Hint: Use when without argument for condition checks [OK]

Common Mistakes:

• Using conditions inside when(variable)
• Omitting variable in range checks
• Misplacing in keyword