LeetCode in Kotlin

929. Unique Email Addresses

Easy

Every valid email consists of a local name and a domain name, separated by the '@' sign. Besides lowercase letters, the email may contain one or more '.' or '+'.

If you add periods '.' between some characters in the local name part of an email address, mail sent there will be forwarded to the same address without dots in the local name. Note that this rule does not apply to domain names.

If you add a plus '+' in the local name, everything after the first plus sign will be ignored. This allows certain emails to be filtered. Note that this rule does not apply to domain names.

It is possible to use both of these rules at the same time.

Given an array of strings emails where we send one email to each emails[i], return the number of different addresses that actually receive mails.

Example 1:

Input: emails = [“test.email+alex@leetcode.com”,”test.e.mail+bob.cathy@leetcode.com”,”testemail+david@lee.tcode.com”]

Output: 2

Explanation: “testemail@leetcode.com” and “testemail@lee.tcode.com” actually receive mails.

Example 2:

Input: emails = [“a@leetcode.com”,”b@leetcode.com”,”c@leetcode.com”]

Output: 3

Constraints:

Solution

class Solution {
    fun numUniqueEmails(emails: Array<String>): Int {
        val set: MutableSet<String> = HashSet()
        for (s in emails) {
            val sb = StringBuilder()
            var i = 0
            while (i < s.length) {
                val c = s[i]
                if (c == '+' || c == '@') {
                    sb.append('@')
                    i = s.indexOf("@") + 1
                    sb.append(s.substring(i))
                    break
                } else if (c != '.') {
                    sb.append(c)
                }
                i++
            }
            set.add(sb.toString())
        }
        return set.size
    }
}