构造限制重复的字符串

Author: yi-ge

.vscode/settings.json

@@ -147,6 +147,7 @@
     "aaabaaa",
     "aaabbaaa",
     "AAABBC",
+    "aababab",
     "aabb",
     "aabcbc",
     "aadsfasf",
@@ -174,11 +175,14 @@
     "backgrounding",
     "Barcodes",
     "bbab",
+    "bbabaa",
+    "bbabaaa",
     "bdca",
     "binutils",
     "breakpad",
     "cadsfafs",
     "camelcase",
+    "cczazcc",
     "Choco",
     "clangd",
     "cmake",
@@ -263,6 +267,8 @@
     "XRLXXRRLX",
     "XRLXXRRXL",
     "XRXLRXRXL",
-    "zaaaz"
+    "zaaaz",
+    "zzcccac",
+    "zzcccca"
   ]
 }

README.md

@@ -152,6 +152,10 @@ C++标准库提供的数据结构实在是太多了，参考[C++标准库头文
 
 ### 字符串
 
+- [构造限制重复的字符串](src/string/construct_string_with_repeat_limit.cpp)  [贪心, 字符串, 计数, 堆（优先队列）]
+
+  - LeetCode 2182. 构造限制重复的字符串 <https://leetcode.cn/problems/construct-string-with-repeat-limit>
+
 - [统计重复个数](src/string/count_the_repetitions.cpp)  [字符串, 动态规划]
 
   - LeetCode 466. 统计重复个数 <https://leetcode.cn/problems/count-the-repetitions>

images/string/construct_string_with_repeat_limit.jpeg

@@ -0,0 +1,37 @@
+// 构造限制重复的字符串
+// https://leetcode.cn/problems/construct-string-with-repeat-limit
+// INLINE  ../../images/string/construct_string_with_repeat_limit.jpeg
+// 参考官方题解
+
+#include <headers.hpp>
+
+class Solution {
+public:
+  string repeatLimitedString(string s, int repeatLimit) {
+    int N = 26;
+    vector<int> count(N);
+    for (char c : s) {
+      count[c - 'a']++;
+    }
+    string ret;
+    int m = 0;
+    for (int i = N - 1, j = N - 2; i >= 0 && j >= 0;) {
+      if (count[i] == 0) { // 当前字符已经填完，填入后面的字符，重置 m
+        m = 0;
+        i--;
+      } else if (m < repeatLimit) { // 当前字符未超过限制
+        count[i]--;
+        ret.push_back('a' + i);
+        m++;
+      } else if (j >= i ||
+                 count[j] == 0) { // 当前字符已经超过限制，查找可填入的其他字符
+        j--;
+      } else { // 当前字符已经超过限制，填入其他字符，并且重置 m
+        count[j]--;
+        ret.push_back('a' + j);
+        m = 0;
+      }
+    }
+    return ret;
+  }
+};

src/string/construct_string_with_repeat_limit.cpp

@@ -1,4 +1,4 @@
-// 执行编译时间：2024-01-12 11:35:49
+// 执行编译时间：2024-01-13 14:28:48
 #include <gtest/gtest.h>
 #include <lib.hpp>
 

test/lib/lib_test.cpp

@@ -0,0 +1,34 @@
+#include <string/construct_string_with_repeat_limit.cpp>
+
+TEST(构造限制重复的字符串, repeatLimitedString) {
+  Solution solution;
+  // 示例 1：
+  // 输入：s = "cczazcc", repeatLimit = 3
+  // 输出："zzcccac"
+  // 解释：使用 s 中的所有字符来构造 repeatLimitedString "zzcccac"。
+  // 字母 'a' 连续出现至多 1 次。
+  // 字母 'c' 连续出现至多 3 次。
+  // 字母 'z' 连续出现至多 2 次。
+  // 因此，没有字母连续出现超过 repeatLimit 次，字符串是一个有效的
+  // repeatLimitedString 。 该字符串是字典序最大的 repeatLimitedString
+  // ，所以返回 "zzcccac" 。 注意，尽管 "zzcccca" 字典序更大，但字母 'c'
+  // 连续出现超过 3 次，所以它不是一个有效的 repeatLimitedString 。
+  string s = "cczazcc";
+  int repeatLimit = 3;
+  EXPECT_EQ(solution.repeatLimitedString(s, repeatLimit), "zzcccac");
+
+  // 示例 2：
+  // 输入：s = "aababab", repeatLimit = 2
+  // 输出："bbabaa"
+  // 使用 s 中的一些字符来构造 repeatLimitedString "bbabaa"。
+  // 解释：
+  // 字母 'a' 连续出现至多 2 次。
+  // 字母 'b' 连续出现至多 2 次。
+  // 因此，没有字母连续出现超过 repeatLimit 次，字符串是一个有效的
+  // repeatLimitedString 。 该字符串是字典序最大的 repeatLimitedString
+  // ，所以返回 "bbabaa" 。 注意，尽管 "bbabaaa" 字典序更大，但字母 'a'
+  // 连续出现超过 2 次，所以它不是一个有效的 repeatLimitedString 。
+  s = "aababab";
+  repeatLimit = 2;
+  EXPECT_EQ(solution.repeatLimitedString(s, repeatLimit), "bbabaa");
+}