Variable-length arguments (*args) in Python - Time & Space Complexity

Time Complexity: Variable-length arguments (*args)

O(n)

Understanding Time Complexity

When using variable-length arguments, the function can take many inputs. We want to know how the time to run changes as the number of inputs grows.

How does the function's work increase when we add more arguments?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

def sum_all(*args):
    total = 0
    for num in args:
        total += num
    return total

This function adds up all the numbers given as arguments.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Looping through each argument in args to add it.
• How many times: Once for each argument passed to the function.

How Execution Grows With Input

As we add more numbers to add, the function does more work, going through each number once.

Input Size (n)	Approx. Operations
10	About 10 additions
100	About 100 additions
1000	About 1000 additions

Pattern observation: The work grows directly with the number of inputs.

Final Time Complexity

Time Complexity: O(n)

This means the time to add all numbers grows in a straight line as you add more numbers.

Common Mistake

[X] Wrong: "Using *args makes the function run instantly no matter how many inputs."

[OK] Correct: Even with *args, the function still needs to look at each input once, so more inputs mean more work.

Interview Connect

Understanding how variable inputs affect time helps you explain your code clearly and shows you think about efficiency, a skill valued in many coding situations.

Self-Check

"What if we changed the function to multiply all arguments instead of adding? How would the time complexity change?"

Practice

1. What does *args do in a Python function?

easy

A. Allows the function to accept any number of extra positional arguments

B. Allows the function to accept only keyword arguments

C. Restricts the function to accept exactly one argument

D. Makes the function return multiple values

Solution

1. Step 1: Understand the role of *args

   *args collects extra positional arguments passed to a function into a tuple.

2. Step 2: Compare options with this behavior

   Only Allows the function to accept any number of extra positional arguments correctly describes this behavior; others describe different concepts.

3. Final Answer:

   Allows the function to accept any number of extra positional arguments -> Option A

4. Quick Check:

   *args = flexible positional inputs [OK]

Hint: Remember *args packs extra positions into a tuple [OK]

Common Mistakes:

• Confusing *args with **kwargs
• Thinking *args limits arguments
• Assuming *args returns multiple values

2. Which of the following is the correct syntax to define a function that accepts variable-length positional arguments?

easy

A. def func(args*):

B. def func(*args):

C. def func(**args):

D. def func(args**):

Solution

1. Step 1: Recall the syntax for variable-length positional arguments

   The correct syntax uses an asterisk before the parameter name: *args.

2. Step 2: Match options with correct syntax

   def func(*args): matches the correct syntax; others are invalid or for different purposes.

3. Final Answer:

   def func(*args): -> Option B

4. Quick Check:

   Star before name = variable positional args [OK]

Hint: Use * before parameter name for variable positional args [OK]

Common Mistakes:

• Using **args instead of *args
• Placing * after the parameter name
• Confusing syntax with keyword arguments

3. What is the output of the following code?

def add_numbers(*args):
    return sum(args)

print(add_numbers(1, 2, 3, 4))

medium

A. 10

B. 1234

C. TypeError

D. None

Solution

1. Step 1: Understand how *args works in the function

   The function collects all arguments into a tuple and sums them using sum().

2. Step 2: Calculate the sum of the arguments

   1 + 2 + 3 + 4 = 10, so the function returns 10.

3. Final Answer:

   10 -> Option A

4. Quick Check:

   Sum of (1,2,3,4) = 10 [OK]

Hint: Sum all *args values to get total [OK]

Common Mistakes:

• Concatenating numbers as strings
• Expecting a TypeError for multiple args
• Forgetting that *args is a tuple

4. Identify the error in this function definition:

def greet(*names):
    for name in names
        print(f"Hello, {name}!")

medium

A. print statement syntax error

B. Incorrect use of *args syntax

C. Missing colon after for loop

D. Function cannot have *args

Solution

1. Step 1: Check the for loop syntax

   The for loop line is missing a colon at the end, which is required in Python.

2. Step 2: Verify other parts of the function

   The *names syntax and print statement are correct; the function can have *args.

3. Final Answer:

   Missing colon after for loop -> Option C

4. Quick Check:

   For loops need a colon [:] [OK]

Hint: Check colons after for, if, while statements [OK]

Common Mistakes:

• Forgetting colon after for loop
• Confusing *args with **kwargs
• Incorrect indentation

5. How can you write a function that accepts any number of positional arguments and returns a list of their squares?

hard

A. def squares(args): return [x**2 for x in args]

B. def squares(*args): return x**2 for x in args

C. def squares(**args): return [x**2 for x in args]

D. def squares(*args): return [x**2 for x in args]

Solution

1. Step 1: Use *args to accept variable positional arguments

   The function parameter must be *args to accept any number of positional arguments.

2. Step 2: Return a list of squares using list comprehension

   Use [x**2 for x in args] to square each argument and collect results in a list.

3. Final Answer:

   def squares(*args): return [x**2 for x in args] -> Option D

4. Quick Check:

   Use *args and list comprehension for squares [OK]

Hint: Use *args and list comprehension for flexible squares [OK]

Common Mistakes:

• Using **args instead of *args
• Forgetting to return a list
• Incorrect comprehension syntax