[Swift]LeetCode692. 前K个高频单词 | Top K Frequent Words

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址： https://www.cnblogs.com/strengthen/p/10502182.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Given a non-empty list of words, return the k most frequent elements.

Your answer should be sorted by frequency from highest to lowest. If two words have the same frequency, then the word with the lower alphabetical order comes first.

Example 1:

Input: ["i", "love", "leetcode", "i", "love", "coding"], k = 2
Output: ["i", "love"]
Explanation: "i" and "love" are the two most frequent words.
    Note that "i" comes before "love" due to a lower alphabetical order. 

Example 2:

Input: ["the", "day", "is", "sunny", "the", "the", "the", "sunny", "is", "is"], k = 4
Output: ["the", "is", "sunny", "day"]
Explanation: "the", "is", "sunny" and "day" are the four most frequent words,
    with the number of occurrence being 4, 3, 2 and 1 respectively. 

Note:

1. You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
2. Input words contain only lowercase letters. 

Follow up:

1. Try to solve it in O(n log k) time and O(n) extra space.

---

给一非空的单词列表，返回前 k 个出现次数最多的单词。

返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率，按字母顺序排序。

示例 1：

输入: ["i", "love", "leetcode", "i", "love", "coding"], k = 2
输出: ["i", "love"]
解析: "i" 和 "love" 为出现次数最多的两个单词，均为2次。
    注意，按字母顺序 "i" 在 "love" 之前。 

示例 2：

输入: ["the", "day", "is", "sunny", "the", "the", "the", "sunny", "is", "is"], k = 4
输出: ["the", "is", "sunny", "day"]
解析: "the", "is", "sunny" 和 "day" 是出现次数最多的四个单词，
    出现次数依次为 4, 3, 2 和 1 次。 

注意：

1. 假定 k 总为有效值， 1 ≤ k ≤ 集合元素数。
2. 输入的单词均由小写字母组成。 

扩展练习：

1. 尝试以 O(n log k) 时间复杂度和 O(n) 空间复杂度解决。

---

76ms

class Solution {
    func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        var frequencyList = [String:Int]()
        for word in words {
            if let count = frequencyList[word] {
                frequencyList[word] = count + 1
            } else {
                frequencyList[word] = 1
            }
        }
        let b = frequencyList.sorted { (dic1, dic2) -> Bool in
            if dic1.value != dic2.value {
                return dic1.value > dic2.value
            } else {
                return dic1.key < dic2.key
            }
        }
        var result = [String]()
        for item in b {
            result.append(item.key)
            if result.count == k {
                return result
            }
        }
        return result
    }
}

---

Runtime: 88 ms

Memory Usage: 20.4 MB

class Solution {
    func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        var k = k
        var res:[String] = [String]()
        var freq:[String:Int] = [String:Int]()
        var v:[Set<String>] = [Set<String>](repeating:Set<String>(),count:words.count + 1)
        for word in words
        {
            freq[word,default:0] += 1
        }
        for (key,val) in freq
        {
            v[val].insert(key)
        }
        for i in stride(from:v.count - 1,through:0,by:-1)
        {
            if k <= 0 {break}
            var t:[String] = v[i].sorted()
            if k >= t.count
            {               
                res += t
            }
            else
            {

                 res += t[0..<k]
            }
            k -= t.count
        }
        return res
    }
}

---

88ms

class Solution {
    func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        var countMap = [String: Int]()
        var firstAppearance = [String: Int]()
        for (i, word) in words.enumerated() {
            countMap[word, default: 0] += 1 
            if firstAppearance[word] == nil {
                firstAppearance[word] = i
            }
        }
        var sortedStringsBasedOnCount = countMap.sorted {
            return $0.value > $1.value || 
            ($0.value == $1.value && $0.key < $1.key)
        }
        return sortedStringsBasedOnCount[0..<k].map { $0.0 }
    }
}

---

92ms

class Solution {
    func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        var countMap = [String: Int]()
        words.forEach { countMap[$0, default: 0] += 1 }
        var sortedStringsBasedOnCount = countMap.sorted {
            return $0.value > $1.value || 
            ($0.value == $1.value && $0.key < $1.key)
        }
        return sortedStringsBasedOnCount[0..<k].map { $0.0 }
    }
}

---

96ms

class Solution {
    func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        var wordDict = [String : Int]()
        for word in words {
            wordDict[word] = 1 + (wordDict[word] ?? 0)
        }
        return wordDict.sorted { 
            $0.value > $1.value || $0.value == $1.value && $0.key < $1.key
        }[0..<k].map{$0.key}
    }
}

---

100ms

class Solution {
    func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        var frequencyList = [String:Int]()
        for word in words {
            if let count = frequencyList[word] {
                frequencyList[word] = count + 1
            } else {
                frequencyList[word] = 1
            }
        }
        var frequencyList2 = [[String]](repeating:[String](), count:words.count)
        for (key, value) in frequencyList {
            frequencyList2[value].append(key)
            frequencyList2[value] = frequencyList2[value].sorted()
        }
        frequencyList2 = frequencyList2.filter{$0 != [String]()}
        var result = [String]()
        var index = frequencyList2.count - 1
        while index >= 0 {
            for item in frequencyList2[index] {
                result.append(item)
                if result.count == k {
                    return result
                }
            }
            index -= 1
        }
        return result
    }
}

---

128ms

class Solution {
       func topKFrequent(_ words: [String], _ k: Int) -> [String] {
        guard words.count > 0 else {
            return [String]()
        }
        
        var memo = [String:Int]()
        
        for w in words {
            memo[w] = memo[w, default:0] + 1
        }
        
        var helpArray = [[String]]()
        
        for key in memo.keys {
            if let count = memo[key] {
                helpArray.append([String(count), key])
            }
        }
        
        helpArray.sort(by:{
            if Int($0[0])! > Int($1[0])! {
                return true
            } else if Int($0[0])! < Int($1[0])! {
                return false
            } else {
                return $0[1] < $1[1]
            }
        })        
        var sortedArray = helpArray.map{ $0[1] }
        
        return Array(sortedArray[0...k - 1])
    }
}