1487-保证文件名唯一

Raphael Liu Lv10

给你一个长度为 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,如果文件系统中不存在名为 name 的文件夹,那么直接创建即可,否则我们需要从 k=1 开始,尝试使用添加后缀 k 的新文件名创建新文件夹。

使用哈希表 index 记录已创建的文件夹的下一后缀序号,遍历 names 数组,记当前遍历的文件名为 name:

  • 如果 name 不在哈希表中,那么说明文件系统不存在名为 name 的文件夹,我们直接创建该文件夹,并且记录对应文件夹的下一后缀序号为 1。

  • 如果 name 在哈希表中,那么说明文件系统已经存在名为 name 的文件夹,我们在哈希表找到 name 的下一后缀序号 k,逐一尝试直到添加后缀 k 的新文件名不存在于哈希表中,然后创建该文件夹。需要注意的是,创建该文件夹后,有两个文件名的下一后缀序号需要修改,首先文件名 name 的下一后缀序号为 k+1,其次,文件名 name 添加后缀 k 的新文件名的下一后缀序号为 1。

[sol1-Python3]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution:
def getFolderNames(self, names: List[str]) -> List[str]:
ans = []
index = {}
for name in names:
if name not in index:
ans.append(name)
index[name] = 1
else:
k = index[name]
while name + '(' + str(k) + ')' in index:
k += 1
t = name + '(' + str(k) + ')'
ans.append(t)
index[name] = k + 1
index[t] = 1
return ans
[sol1-C++]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
class Solution {
public:
string addSuffix(string name, int k) {
return name + "(" + to_string(k) + ")";
}

vector<string> getFolderNames(vector<string>& names) {
unordered_map<string, int> index;
vector<string> res;
for (const auto &name : names) {
if (!index.count(name)) {
res.push_back(name);
index[name] = 1;
} else {
int k = index[name];
while (index.count(addSuffix(name, k))) {
k++;
}
res.push_back(addSuffix(name, k));
index[name] = k + 1;
index[addSuffix(name, k)] = 1;
}
}
return res;
}
};
[sol1-Java]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
public String[] getFolderNames(String[] names) {
Map<String, Integer> index = new HashMap<String, Integer>();
int n = names.length;
String[] res = new String[n];
for (int i = 0; i < n; i++) {
String name = names[i];
if (!index.containsKey(name)) {
res[i] = name;
index.put(name, 1);
} else {
int k = index.get(name);
while (index.containsKey(addSuffix(name, k))) {
k++;
}
res[i] = addSuffix(name, k);
index.put(name, k + 1);
index.put(addSuffix(name, k), 1);
}
}
return res;
}

public String addSuffix(String name, int k) {
return name + "(" + k + ")";
}
}
[sol1-C#]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
public class Solution {
public string[] GetFolderNames(string[] names) {
IDictionary<string, int> index = new Dictionary<string, int>();
int n = names.Length;
string[] res = new string[n];
for (int i = 0; i < n; i++) {
string name = names[i];
if (!index.ContainsKey(name)) {
res[i] = name;
index.Add(name, 1);
} else {
int k = index[name];
while (index.ContainsKey(AddSuffix(name, k))) {
k++;
}
res[i] = AddSuffix(name, k);
index[name] = k + 1;
index.Add(AddSuffix(name, k), 1);
}
}
return res;
}

public string AddSuffix(string name, int k) {
return name + "(" + k + ")";
}
}
[sol1-C]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
typedef struct {
char *key;
int val;
UT_hash_handle hh;
} HashItem;

HashItem *hashFindItem(HashItem **obj, char *key) {
HashItem *pEntry = NULL;
HASH_FIND_STR(*obj, key, pEntry);
return pEntry;
}

bool hashAddItem(HashItem **obj, char *key, int val) {
if (hashFindItem(obj, key)) {
return false;
}
HashItem *pEntry = (HashItem *)malloc(sizeof(HashItem));
pEntry->key = (char *)calloc(sizeof(char), strlen(key) + 1);
strcpy(pEntry->key, key);
pEntry->val = val;
HASH_ADD_STR(*obj, key, pEntry);
return true;
}

bool hashSetItem(HashItem **obj, const char *key, int val) {
HashItem *pEntry = hashFindItem(obj, key);
if (!pEntry) {
hashAddItem(obj, key, val);
} else {
pEntry->val = val;
}
return true;
}

int hashGetItem(HashItem **obj, char *key, int defaultVal) {
HashItem *pEntry = hashFindItem(obj, key);
if (!pEntry) {
return defaultVal;
}
return pEntry->val;
}

void hashFree(HashItem **obj) {
HashItem *curr = NULL, *tmp = NULL;
HASH_ITER(hh, *obj, curr, tmp) {
HASH_DEL(*obj, curr);
free(curr->key);
free(curr);
}
}

char ** getFolderNames(char ** names, int namesSize, int* returnSize) {
HashItem *index = NULL;
char **res = (char **)calloc(sizeof(char *), namesSize);
int pos = 0;
for (int i = 0; i < namesSize; i++) {
if (!hashFindItem(&index, names[i])) {
res[pos] = (char *)calloc(sizeof(char), strlen(names[i]) + 1);
strcpy(res[pos], names[i]);
pos++;
hashAddItem(&index, names[i], 1);
} else {
int k = hashGetItem(&index, names[i], 0);
char str[strlen(names[i]) + 16];
sprintf(str, "%s(%d)", names[i], k);
while (hashFindItem(&index, str)) {
k++;
sprintf(str, "%s(%d)", names[i], k);
}
res[pos] = (char *)calloc(sizeof(char), strlen(str) + 1);
strcpy(res[pos], str);
pos++;
hashSetItem(&index, names[i], k + 1);
hashAddItem(&index, str, 1);
}
}
*returnSize = pos;
hashFree(&index);
return res;
}
[sol1-JavaScript]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
var getFolderNames = function(names) {
const index = new Map();
const n = names.length;
const res = new Array(n).fill(0);
for (let i = 0; i < n; i++) {
const name = names[i];
if (!index.has(name)) {
res[i] = name;
index.set(name, 1);
} else {
let k = index.get(name);
while (index.has(addSuffix(name, k))) {
k++;
}
res[i] = addSuffix(name, k);
index.set(name, k + 1);
index.set(addSuffix(name, k), 1);
}
}
return res;
}

const addSuffix = (name, k) => {
return name + "(" + k + ")";
};
[sol1-Golang]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
func getFolderNames(names []string) []string {
ans := make([]string, len(names))
index := map[string]int{}
for p, name := range names {
i := index[name]
if i == 0 {
index[name] = 1
ans[p] = name
continue
}
for index[name+"("+strconv.Itoa(i)+")"] > 0 {
i++
}
t := name + "(" + strconv.Itoa(i) + ")"
ans[p] = t
index[name] = i + 1
index[t] = 1
}
return ans
}

复杂度分析

  • 时间复杂度:O(\sum_{i=0}^{n-1} m_i),其中 m_i 表示字符串 names}[i] 的长度,n 表示数组 names 的长度。外层循环每次运行时间为 O(m_i),总时间复杂度为 O(\sum_{i=0}^{n-1} m_i);内层循环的总运行次数不超过 n 次,总时间复杂度为 O(\sum_{i=0}^{n-1} m_i)。

  • 空间复杂度:O(\sum_{i=0}^{n-1}{m_i})。保存哈希表需要 O(\sum_{i=0}^{n-1}{m_i}) 的空间。

 Comments
On this page
1487-保证文件名唯一