Start learning free

LESSON 383 OF 415 · PROJECTS: COMMAND-LINE TOOLS

CLI Tool: wc v2 — Count Words in Swift

Practice

Build a tiny version of the Unix wc -w command. Given a list of lines, return the total number of words (whitespace-separated tokens) across all lines as an integer.

Function signature:

func countWords(_ lines: [String]) -> Int

Example:

Input: lines = ["hello world", "foo bar baz", ""]

Output: 5

The empty string contributes 0 words, "hello world" contributes 2, and "foo bar baz" contributes 3. Total: 5.

Words are separated by one or more spaces or tabs, and leading or trailing whitespace is ignored. A line with no words (empty, or only whitespace) contributes 0, and an empty list of lines gives 0. The input is plain ASCII text.


Iterate over the lines and accumulate the word count. Splitting on runs of whitespace naturally drops empty pieces, so blank or whitespace-only lines contribute zero.

Split each line on whitespace and count the tokens. split(whereSeparator:) collapses consecutive whitespace and omits empty pieces, so empty lines add nothing:
total += line.split(whereSeparator: { $0.isWhitespace }).count


The full solution:
class Solution {
    func countWords(_ lines: [String]) -> Int {
        var total = 0
        for line in lines {
            total += line.split(whereSeparator: { $0.isWhitespace }).count
        }
        return total
    }
}

STEP 1 OF 4
    Ctrl + Enter to check

    Wc V2 in Swift

    Keep going with SwiftAbout 340 step-by-step lessons, from your first line of code to coding interviews. Free to start.
    Start learning free