LeetCode in Kotlin

884. Uncommon Words from Two Sentences

Easy

A sentence is a string of single-space separated words where each word consists only of lowercase letters.

A word is uncommon if it appears exactly once in one of the sentences, and does not appear in the other sentence.

Given two sentences s1 and s2, return a list of all the uncommon words. You may return the answer in any order.

Example 1:

Input: s1 = “this apple is sweet”, s2 = “this apple is sour”

Output: [“sweet”,”sour”]

Example 2:

Input: s1 = “apple apple”, s2 = “banana”

Output: [“banana”]

Constraints:

Solution

class Solution {
    fun uncommonFromSentences(s1: String, s2: String): Array<String?> {
        val visited = HashSet<String>()
        val uniques = HashSet<String>()
        for (word in s1.split(" ".toRegex()).dropLastWhile { it.isEmpty() }.toTypedArray()) {
            if (!visited.add(word)) {
                uniques.remove(word)
            } else {
                uniques.add(word)
            }
        }
        for (word in s2.split(" ".toRegex()).dropLastWhile { it.isEmpty() }.toTypedArray()) {
            if (!visited.add(word)) {
                uniques.remove(word)
            } else {
                uniques.add(word)
            }
        }
        val arr = arrayOfNulls<String>(uniques.size)
        for ((i, word) in uniques.withIndex()) {
            arr[i] = word
        }
        return arr
    }
}