Start learning free

LESSON 615 OF 678 · PROJECTS: COMMAND-LINE TOOLS

CLI Tool: wc v2 — Count Words

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