题目描述
你有一个 9 键键盘,按键按 1 到 9 编号,每个按键对应着几个英文小写字母。你可以决定每个按键对应哪些英文字母,但要满足如下条件:
- 26 个英文小写字母必须全部映射到这 9 个按键上。
- 每个英文字母只能映射到 恰好 一个按键上。
- 每个按键 最多 对应 3 个英文字母。
如果想打出按键上的第一个字母,只需要按一次。如果想打出按键上的第二个字母,则需要按两次,依次类推。
给你一个字符串 s
,返回基于你设计的键盘打出 s
需要的 最少 按键次数。
注意:字母映射到每个按键上,映射的顺序无法进行更改。
示例 1 :
输入:s = "apple"
输出:5
解释:上图所示为设置键盘的最佳方法之一。
按按键 1 一次输入 'a' 。
按按键 6 一次输入 'p' 。
按按键 6 一次输入 'p' 。
按按键 5 一次输入 'l' 。
按按键 3 一次输入 'e' 。
总共按按键 5 次,所以返回 5 。
示例 2 :
输入:s = "abcdefghijkl"
输出:15
解释:上图所示为设置键盘的最佳方法之一。
字母 'a' 到 'i' 每个只需要按一次按键。
按按键 1 两次输入 'j' 。
按按键 2 两次输入 'k' 。
按按键 3 两次输入 'l' 。
总共按按键 15 次,所以返回 15 。
提示:
1 <= s.length <= 105
s
由小写英文字母组成
解法
方法一:计数 + 贪心
我们首先统计字符串 $s$ 中每个字符出现的次数,记录在数组或者哈希表 $\textit{cnt}$ 中。
题目要求按键次数最少,那么出现最多的 $9$ 个字符应该对应按键 $1$ 到按键 $9$,出现次数第 $10$ 到第 $18$ 多的字符再次对应按键 $1$ 到按键 $9$,以此类推。
因此,我们可以将 $\textit{cnt}$ 中的值按照从大到小的顺序排序,然后按照 $1$ 到 $9$ 的顺序依次分配给按键,每次分配完 $9$ 个字符后,按键次数加 $1$。
时间复杂度 $O(n + |\Sigma| \times \log |\Sigma|)$,空间复杂度 $O(|\Sigma|)$。其中 $n$ 是字符串 $s$ 的长度,而 $\Sigma$ 是字符串 $s$ 中出现的字符集合,本题中 $\Sigma$ 是小写字母集合,因此 $|\Sigma| = 26$。
| class Solution:
def minimumKeypresses(self, s: str) -> int:
cnt = Counter(s)
ans, k = 0, 1
for i, x in enumerate(sorted(cnt.values(), reverse=True), 1):
ans += k * x
if i % 9 == 0:
k += 1
return ans
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 | class Solution {
public int minimumKeypresses(String s) {
int[] cnt = new int[26];
for (int i = 0; i < s.length(); ++i) {
++cnt[s.charAt(i) - 'a'];
}
Arrays.sort(cnt);
int ans = 0, k = 1;
for (int i = 1; i <= 26; ++i) {
ans += k * cnt[26 - i];
if (i % 9 == 0) {
++k;
}
}
return ans;
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 | class Solution {
public:
int minimumKeypresses(string s) {
int cnt[26]{};
for (char& c : s) {
++cnt[c - 'a'];
}
sort(begin(cnt), end(cnt), greater<int>());
int ans = 0, k = 1;
for (int i = 1; i <= 26; ++i) {
ans += k * cnt[i - 1];
if (i % 9 == 0) {
++k;
}
}
return ans;
}
};
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 | func minimumKeypresses(s string) (ans int) {
cnt := make([]int, 26)
for _, c := range s {
cnt[c-'a']++
}
sort.Ints(cnt)
k := 1
for i := 1; i <= 26; i++ {
ans += k * cnt[26-i]
if i%9 == 0 {
k++
}
}
return
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 | function minimumKeypresses(s: string): number {
const cnt: number[] = Array(26).fill(0);
const a = 'a'.charCodeAt(0);
for (const c of s) {
++cnt[c.charCodeAt(0) - a];
}
cnt.sort((a, b) => b - a);
let [ans, k] = [0, 1];
for (let i = 1; i <= 26; ++i) {
ans += k * cnt[i - 1];
if (i % 9 === 0) {
++k;
}
}
return ans;
}
|