comments | difficulty | edit_url | rating | source | tags | |||
---|---|---|---|---|---|---|---|---|
true |
中等 |
1696 |
第 194 场周赛 Q2 |
|
给你一个长度为 n
的字符串数组 names
。你将会在文件系统中创建 n
个文件夹:在第 i
分钟,新建名为 names[i]
的文件夹。
由于两个文件 不能 共享相同的文件名,因此如果新建文件夹使用的文件名已经被占用,系统会以 (k)
的形式为新文件夹的文件名添加后缀,其中 k
是能保证文件名唯一的 最小正整数 。
返回长度为 n
的字符串数组,其中 ans[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 <= names.length <= 5 * 10^4
1 <= names[i].length <= 20
names[i]
由小写英文字母、数字和/或圆括号组成。
我们可以用哈希表
接下来遍历文件夹数组,对于每个文件名
- 如果
$name$ 在$d$ 中,说明文件夹$name$ 已经存在,我们需要找到一个新的文件夹名字。我们可以不断地尝试$name(k)$ ,其中$k$ 从$d[name]$ 开始,直到找到一个文件夹名字$name(k)$ 不存在于$d$ 中为止。我们将$name(k)$ 加入$d$ 中,并将$d[name]$ 更新为$k + 1$ 。然后我们将$name$ 更新为$name(k)$ 。 - 如果
$name$ 不在$d$ 中,我们可以直接将$name$ 加入$d$ 中,并将$d[name]$ 更新为$1$ 。 - 接着我们将
$name$ 加入答案数组。然后继续遍历下一个文件名。
遍历完所有文件名后,我们即可得到答案数组。
在以下代码实现中,我们直接修改文件名数组
$names$ ,而不使用额外的答案数组。
时间复杂度
class Solution:
def getFolderNames(self, names: List[str]) -> List[str]:
d = defaultdict(int)
for i, name in enumerate(names):
if name in d:
k = d[name]
while f'{name}({k})' in d:
k += 1
d[name] = k + 1
names[i] = f'{name}({k})'
d[names[i]] = 1
return names
class Solution {
public String[] getFolderNames(String[] names) {
Map<String, Integer> d = new HashMap<>();
for (int i = 0; i < names.length; ++i) {
if (d.containsKey(names[i])) {
int k = d.get(names[i]);
while (d.containsKey(names[i] + "(" + k + ")")) {
++k;
}
d.put(names[i], k);
names[i] += "(" + k + ")";
}
d.put(names[i], 1);
}
return names;
}
}
class Solution {
public:
vector<string> getFolderNames(vector<string>& names) {
unordered_map<string, int> d;
for (auto& name : names) {
int k = d[name];
if (k) {
while (d[name + "(" + to_string(k) + ")"]) {
k++;
}
d[name] = k;
name += "(" + to_string(k) + ")";
}
d[name] = 1;
}
return names;
}
};
func getFolderNames(names []string) []string {
d := map[string]int{}
for i, name := range names {
if k, ok := d[name]; ok {
for {
newName := fmt.Sprintf("%s(%d)", name, k)
if d[newName] == 0 {
d[name] = k + 1
names[i] = newName
break
}
k++
}
}
d[names[i]] = 1
}
return names
}
function getFolderNames(names: string[]): string[] {
let d: Map<string, number> = new Map();
for (let i = 0; i < names.length; ++i) {
if (d.has(names[i])) {
let k: number = d.get(names[i]) || 0;
while (d.has(names[i] + '(' + k + ')')) {
++k;
}
d.set(names[i], k);
names[i] += '(' + k + ')';
}
d.set(names[i], 1);
}
return names;
}