LC 899. 有序队列
题目描述
这是 LeetCode 上的 899. 有序队列 ,难度为 困难。
给定一个字符串 s
和一个整数 k
。
你可以从 s
的前 k
个字母中选择一个,并把它加到字符串的末尾。
返回 在应用上述步骤的任意数量的移动后,字典上最小的字符串 。
示例 1:1
2
3
4
5
6
7输入:s = "cba", k = 1
输出:"acb"
解释:
在第一步中,我们将第一个字符(“c”)移动到最后,获得字符串 “bac”。
在第二步中,我们将第一个字符(“b”)移动到最后,获得最终结果 “acb”。
示例 2:1
2
3
4
5
6
7输入:s = "baaca", k = 3
输出:"aaabc"
解释:
在第一步中,我们将第一个字符(“b”)移动到最后,获得字符串 “aacab”。
在第二步中,我们将第三个字符(“c”)移动到最后,获得最终结果 “aaabc”。
提示:
- $1 <= k <= S.length <= 1000$
s
只由小写字母组成。
最小表示法
当 $k > 1$ 时,我们能够构造出任意的字符串方案,因此当 $k > 1$ 时,我们可以直接通过对字符串排序来得到答案,复杂度为 $O(n\log{n})$。
当 $k = 1$ 时,我们共有 $n$ 种候选方案(将字符串 s
看作一个首尾相接的循环字符串,共有 $n$ 个起点可枚举),枚举过程中需要与当前最优的方案进行比较,比较复杂度为 $O(n)$,因此整体复杂度为 $O(n^2)$。
上述的做法已经可以通过本题,可以看出瓶颈在于对 $k = 1$ 的处理。
而实际上,对于给定字符串 s
,求其循环同构的所有方案中字典序最小的方案,可以使用「最小表示法」来做,复杂度为 $O(n)$。
最小表示法将「方案比较」与「构造更优方案」进行结合:假设我们当前有两字符串 a
和 b
需要进行比较,其均为原串 s
的循环同构具体方案。假设 a
和 b
分别对应了原串下标为 i
和 j
的具体方案,且假设两字符串前 $k$ 个字符均相同。
当两字符串第一个不同的字符大小关系为 $cs[i + k] > cs[j + k]$ 时,可以发现在下标范围 $idx \in [i, i + k]$ 作为起点的新方案 a'
必然不会是最优方案,即必然存在下标范围 $idx - i + j$ 作为起点的新方案 b'
比其更优,因此我们可以直接从 $i + k + 1$ 位置构造新的更优方案,并与 b
再次比较。而 $cs[i + k] < cs[j + k]$ 的分析同理。
更为直白的表述为:分别从
i
和j
作为起点的字符串a
和b
,其前 $k$ 个字符相同,而当 $cs[i + k] > cs[j + k]$ 时,我们可以明确「以 $i + p$ 为起点的字符串」必不可能比「以 $j + p$ 为起点的字符串」更优,其中 $p \in [0, k]$。
Java 代码:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23class Solution {
public String orderlyQueue(String s, int _k) {
char[] cs = s.toCharArray();
if (_k == 1) {
int i = 0, j = 1, k = 0, n = cs.length;
while (i < n && j < n && k < n) {
char a = cs[(i + k) % n], b = cs[(j + k) % n];
if (a == b) k++;
else {
if (a > b) i += k + 1;
else j += k + 1;
if (i == j) i++;
k = 0;
}
}
i = Math.min(i, j);
return s.substring(i) + s.substring(0, i);
} else {
Arrays.sort(cs);
return String.valueOf(cs);
}
}
}
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
26class Solution {
public:
string orderlyQueue(string s, int _k) {
if (_k == 1) {
int i = 0, j = 1, k = 0, n = s.length();
while (i < n && j < n && k < n) {
char a = s[(i + k) % n], b = s[(j + k) % n];
if (a == b) k++;
else {
if (a > b) i += k + 1;
else j += k + 1;
if (i == j) i++;
k = 0;
}
}
i = min(i, j);
return s.substr(i) + s.substr(0, i);
} else {
char* cs = new char[s.length()];
copy(s.begin(), s.end(), cs);
sort(cs, cs + s.length());
string sortedString(cs, s.length());
return sortedString;
}
}
};
Python 代码:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20class Solution:
def orderlyQueue(self, s: str, _k: int) -> str:
if _k == 1:
i, j, k, n = 0, 1, 0, len(s)
while i < n and j < n and k < n:
a, b = s[(i + k) % n], s[(j + k) % n]
if a == b:
k += 1
else:
if a > b:
i += k + 1
else:
j += k + 1
if i == j:
i += 1
k = 0
i = min(i, j)
return s[i:] + s[:i]
else:
return ''.join(sorted(s))
TypeScript 代码:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19function orderlyQueue(s: string, _k: number): string {
if (_k == 1) {
let i = 0, j = 1, k = 0, n = s.length
while (i < n && j < n && k < n) {
const a = s[(i + k) % n], b = s[(j + k) % n]
if (a == b) k++;
else {
if (a > b) i += k + 1
else j += k + 1
if (i == j) i++
k = 0
}
}
i = Math.min(i, j)
return s.substring(i) + s.substring(0, i)
} else {
return [...s].sort().join('');
}
};
- 时间复杂度:当 $k = 1$ 时,复杂度为 $O(n)$;当 $k > 1$ 时,复杂度为 $O(n\log{n})$
- 空间复杂度:当 $k > 1$ 时,需要使用额外的排序空间 $O(\log{n})$
最后
这是我们「刷穿 LeetCode」系列文章的第 No.899
篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。
在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。
为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:https://github.com/SharingSource/LogicStack-LeetCode 。
在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!