难度: m i d d l e \color{orange}{middle} middle
题目描述
给你一个长度为 n n n 的字符串数组 n a m e s names names 。你将会在文件系统中创建 n n n 个文件夹:在第 i i i 分钟,新建名为 n a m e s [ i ] names[i] names[i] 的文件夹。
由于两个文件 不能 共享相同的文件名,因此如果新建文件夹使用的文件名已经被占用,系统会以 ( k ) (k) (k) 的形式为新文件夹的文件名添加后缀,其中 k k k 是能保证文件名唯一的 最小正整数 。
返回长度为 n n n 的字符串数组,其中 a n s [ i ] ans[i] ans[i] 是创建第 i i i 个文件夹时系统分配给该文件夹的实际名称。
示例 1:
输入:names = ["pes","fifa","gta","pes(2019)"]
输出:["pes","fifa","gta","pes(2019)"]
解释:文件系统将会这样创建文件名:
"pes" --> 之前未分配,仍为 "pes"
"fifa" --> 之前未分配,仍为 "fifa"
"gta" --> 之前未分配,仍为 "gta"
"pes(2019)" --> 之前未分配,仍为 "pes(2019)"
示例 2:
输入:names = ["gta","gta(1)","gta","avalon"]
输出:["gta","gta(1)","gta(2)","avalon"]
解释:文件系统将会这样创建文件名:
"gta" --> 之前未分配,仍为 "gta"
"gta(1)" --> 之前未分配,仍为 "gta(1)"
"gta" --> 文件名被占用,系统为该名称添加后缀 (k),由于 "gta(1)" 也被占用,所以 k = 2 。实际创建的文件名为 "gta(2)" 。
"avalon" --> 之前未分配,仍为 "avalon"
示例 3:
输入:names = ["onepiece","onepiece(1)","onepiece(2)","onepiece(3)","onepiece"]
输出:["onepiece","onepiece(1)","onepiece(2)","onepiece(3)","onepiece(4)"]
解释:当创建最后一个文件夹时,最小的正有效 k 为 4 ,文件名变为 "onepiece(4)"。
示例 4:
输入:names = ["wano","wano","wano","wano"]
输出:["wano","wano(1)","wano(2)","wano(3)"]
解释:每次创建文件夹 "wano" 时,只需增加后缀中 k 的值即可。
示例 5:
输入:names = ["kaido","kaido(1)","kaido","kaido(1)"]
输出:["kaido","kaido(1)","kaido(2)","kaido(1)(1)"]
解释:注意,如果含后缀文件名被占用,那么系统也会按规则在名称后添加新的后缀 (k) 。
提示:
- 1 < = n a m e s . l e n g t h < = 5 ∗ 1 0 4 1 <= names.length <= 5 * 10^4 1<=names.length<=5∗104
- 1 < = n a m e s [ i ] . l e n g t h < = 20 1 <= names[i].length <= 20 1<=names[i].length<=20
- n a m e s [ i ] names[i] names[i] 由小写英文字母、数字和/或圆括号组成。
算法
(哈希表)
对于需要被创建的文件名 name
,如果文件系统中不存在名为 name
的文件夹,那么直接创建即可,否则我们需要从 k=1
开始,尝试使用添加后缀 k
的新文件名创建新文件夹。
使用哈希表 hash
记录已创建的文件夹的下一后缀序号,遍历 names
数组,记当前遍历的文件名为 name
:
-
如果
name
不在哈希表中,那么说明文件系统不存在名为name
的文件夹,我们直接创建该文件夹,并且记录对应文件夹的下一后缀序号为1
。 -
如果
name
在哈希表中,那么说明文件系统已经存在名为name
的文件夹,我们在哈希表找到name
的下一后缀序号k
,逐一尝试直到添加后缀k
的新文件名不存在于哈希表中,然后创建该文件夹。需要注意的是,创建该文件夹后,有两个文件名的下一后缀序号需要修改,首先文件名name
的下一后缀序号为k+1
,其次,文件名name
添加后缀k
的新文件名的下一后缀序号为1
。
C++ 代码
class Solution {
public:
vector<string> getFolderNames(vector<string>& names) {
unordered_map<string, int> hash;
vector<string> res;
for (auto name : names) {
if (!hash.count(name)) {
res.push_back(name);
hash[name] ++;
} else {
int k = hash[name];
while (hash.count(name + "(" + to_string(k) + ")"))
k ++;
res.push_back(name + "(" + to_string(k) + ")");
hash[name] = k + 1;
hash[name + "(" + to_string(k) + ")"] ++;
}
}
return res;
}
};