2026-07-27:连接二进制片段得到的最大值。用go语言,给定两个长度为 n 的整数数组 nums1 和 nums0,其中 nums1[i] 代表第 i 个片段中 '1' 的个数,nums0[i] 代表该片段中 '0' 的个数。对于每个 i,我们构造一个二进制片段:先写入 nums1[i] 个连续的 '1',紧接着写入 nums0[i] 个连续的 '0'。我们可以将这些片段以任意顺序重新排列,然后将排列后的所有片段依次拼接成一个完整的二进制字符串。要求找出在所有可能的排列方式中,该二进制字符串所能表示的最大整数值。由于答案可能很大,请将其对 1000000007 取模后返回。
1 <= n == nums1.length == nums0.length <= 100000。
0 <= nums1[i], nums0[i] <= 10000。
nums1[i] + nums0[i] > 0。
nums1 和 nums0 中所有元素的总和不超过 200000。
输入: nums1 = [1,2], nums0 = [1,0]。
输出: 14。
解释:
在下标 0 处,nums1[0] = 1 且 nums0[0] = 1,因此形成的片段为 "10"。
在下标 1 处,nums1[1] = 2 且 nums0[1] = 0,因此形成的片段为 "11"。
将片段重新排序为 "11" 后跟 "10",生成二进制字符串 "1110"。
二进制数 "1110" 的值为 14,这是可能的最大值。
题目来自力扣3897。
大体步骤如下:
1. 预处理:计算2的幂次方
- 目的:由于在后续计算中需要频繁地计算
2^k mod MOD(其中k是每个片段中 '1' 或 '0' 的数量),提前预处理好这些值可以避免重复计算,大大提高效率。 - 过程:创建一个大小为
mx(10001)的数组pow2。pow2[0]初始化为 1(代表2^0)。- 通过一个循环,利用递推关系
pow2[i] = (pow2[i-1] * 2) % MOD,计算出从2^1到2^10000的所有值并存储起来。
2. 确定片段的最佳拼接顺序
这是算法的核心。目标是找出一种排列顺序,使得最终拼接成的二进制字符串表示的数值最大。
-
初始化:创建一个索引数组
idx,长度等于片段总数n,并填入0到n-1的序号。这个数组用于后续的排序,我们不是直接移动原始的片段数据,而是对它们的索引进行排序。 -
自定义排序规则:我们需要定义一种比较逻辑,来判断任意两个片段
A和B,谁排在前面能使最终结果更大。这里的比较策略非常巧妙:- 特殊情况处理:比较片段
i和片段j。- 规则1:如果片段
i的 '0' 的个数为 0(nums0[i] == 0),那么这个片段应该排在任何带有 '0' 的片段(nums0[j] > 0)之前。一个纯 '1' 的片段放在前面,可以确保它的高位全是 '1',从而最大化整个数值。 - 规则2(规则1的补充):如果片段
j的 '0' 的个数为 0,那么它应该排在片段i之前。
- 规则1:如果片段
- 一般情况比较:如果两个片段都包含至少一个 '0'(即
nums0[i] > 0且nums0[j] > 0),则我们需要一个通用的比较方法。- 我们实际上是在比较两种拼接方案:
片段i + 片段j和片段j + 片段i,哪个更大。 - 可以证明,这种比较可以转化为优先比较两个片段中 '1' 的个数。具体来看:首先比较
nums1[j]和nums1[i]的差值。如果nums1[j] - nums1[i] != 0,则意味着一个片段的 '1' 比另一个多。拥有更多 '1' 的片段应排在前面,因为它能为高位贡献更多的 '1'。 - 如果两个片段的 '1' 的个数完全相同(
nums1[j] == nums1[i]),那么就需要比较它们 '0' 的个数。此时,包含更少 '0' 的片段应排在前面。因为更少的 '0' 意味着这个片段的结束部分会更短,能更快地过渡到下一个片段的 '1',避免在数值的高位部分留下过多的 '0'。
- 我们实际上是在比较两种拼接方案:
- 排序执行:使用这个复杂的自定义比较规则,对索引数组
idx进行排序。排序后,idx数组中的索引顺序就代表了片段的最佳拼接顺序。
- 特殊情况处理:比较片段
3. 迭代计算最终的最大值
在得到了最佳拼接顺序(即 idx 数组)后,我们模拟拼接过程,逐步计算出最终数值的十进制表示(对 MOD 取模)。
- 初始化答案
ans为 0。 - 按最优顺序遍历片段:依次取出
idx中的索引i。 - 状态转移(核心公式):
- 假设当前已拼接好的前缀字符串对应的数值是
ans。 - 下一个要拼接的片段包含
ones(nums1[i])个 '1' 和zeros(nums0[i])个 '0'。 - 步骤3.1(追加'1's):将当前值
ans左移ones位(相当于乘以2^ones),然后追加ones个 '1'。这ones个 '1' 代表的数值是(2^ones - 1)。因此,这一步操作可以表达为:新值 = ans * (2^ones) + (2^ones - 1)。代码中巧妙地将其合并为(ans + 1) * pow2[ones] - 1。 - 步骤3.2(追加'0's):在步骤3.1的结果后面再追加
zeros个 '0'。这相当于将当前值左移zeros位(相当于乘以2^zeros)。因此,这一步操作表达为:最终新值 = 步骤3.1的结果 * pow2[zeros]。 - 取模:每一步计算
新值时,都对MOD进行取模运算,确保ans不会溢出,且满足题目要求。
- 假设当前已拼接好的前缀字符串对应的数值是
- 完成:遍历完所有片段后,最终的
ans就是所求的最大整数值。
复杂度分析
-
总的时间复杂度:O(n log n + M)
M是mx,即 10001。预处理pow2数组的时间复杂度是 O(M)。- 排序索引数组
idx的时间复杂度是 O(n log n),n是片段的数量。 - 迭代计算最终值的过程是 O(n)。
- 主要瓶颈在于排序,因此总时间复杂度为 O(n log n + M)。
-
总的额外空间复杂度:O(n + M)
pow2数组的大小固定为M(10001),空间复杂度为 O(M)。- 索引数组
idx的长度为n,空间复杂度为 O(n)。 - 其他变量使用的空间是常数级。
- 因此,总的额外空间需求是 O(n + M)。
Go完整代码如下:
package main
import (
"cmp"
"fmt"
"slices"
)
const mod = 1_000_000_007
const mx = 10001
var pow2 = [mx]int{1}
func init() {
// 预处理 2 的幂
for i := 1; i < mx; i++ {
pow2[i] = pow2[i-1] * 2 % mod
}
}
func maxValue(nums1, nums0 []int) (ans int) {
idx := make([]int, len(nums1))
for i := range idx {
idx[i] = i
}
slices.SortFunc(idx, func(i, j int) int {
if nums0[i] == 0 {
return -1
}
if nums0[j] == 0 {
return 1
}
return cmp.Or(nums1[j]-nums1[i], nums0[i]-nums0[j])
})
for _, i := range idx {
ans = ((ans+1)*pow2[nums1[i]] - 1) % mod * pow2[nums0[i]] % mod
}
return
}
func main() {
nums1 := []int{1, 2}
nums0 := []int{1, 0}
result := maxValue(nums1, nums0)
fmt.Println(result)
}
Python完整代码如下:
# -*-coding:utf-8-*-
MOD = 1_000_000_007
MX = 10001
# 预处理 2 的幂
pow2 = [1] * MX
for i in range(1, MX):
pow2[i] = pow2[i-1] * 2 % MOD
def maxValue(nums1, nums0):
n = len(nums1)
idx = list(range(n))
# 自定义排序函数
def sort_key(i):
if nums0[i] == 0:
return (0, 0, 0) # 负数标记,排最前面
if nums0[j] == 0: # 这个在排序比较中无法直接使用,需要改为cmp方式
return (2, 0, 0) # 正数标记,排最后面
# 使用functools.cmp_to_key来实现自定义比较
from functools import cmp_to_key
def cmp_func(i, j):
if nums0[i] == 0:
return -1
if nums0[j] == 0:
return 1
# cmp.Or(nums1[j]-nums1[i], nums0[i]-nums0[j])
diff1 = nums1[j] - nums1[i]
if diff1 != 0:
return diff1
return nums0[i] - nums0[j]
idx.sort(key=cmp_to_key(cmp_func))
ans = 0
for i in idx:
ans = ((ans + 1) * pow2[nums1[i]] - 1) % MOD * pow2[nums0[i]] % MOD
return ans
def main():
nums1 = [1, 2]
nums0 = [1, 0]
result = maxValue(nums1, nums0)
print(result)
if __name__ == "__main__":
main()
C++完整代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
#include <functional>
const int MOD = 1'000'000'007;
const int MX = 10001;
// 预处理 2 的幂
std::vector<int> pow2(MX);
void init() {
pow2[0] = 1;
for (int i = 1; i < MX; i++) {
pow2[i] = (pow2[i-1] * 2LL) % MOD;
}
}
int maxValue(const std::vector<int>& nums1, const std::vector<int>& nums0) {
int n = nums1.size();
std::vector<int> idx(n);
for (int i = 0; i < n; i++) {
idx[i] = i;
}
// 自定义排序
std::sort(idx.begin(), idx.end(), [&](int i, int j) {
if (nums0[i] == 0) {
return true; // i 排在前面
}
if (nums0[j] == 0) {
return false; // j 排在前面
}
// cmp.Or(nums1[j]-nums1[i], nums0[i]-nums0[j])
int diff1 = nums1[j] - nums1[i];
if (diff1 != 0) {
return diff1 < 0; // nums1[i] > nums1[j] 时 i 排在前面
}
return nums0[i] - nums0[j] < 0;
});
long long ans = 0;
for (int i : idx) {
ans = (((ans + 1) * pow2[nums1[i]] - 1) % MOD) * pow2[nums0[i]] % MOD;
ans = (ans + MOD) % MOD; // 确保结果为正
}
return static_cast<int>(ans);
}
int main() {
// 初始化pow2数组
init();
std::vector<int> nums1 = {1, 2};
std::vector<int> nums0 = {1, 0};
int result = maxValue(nums1, nums0);
std::cout << result << std::endl;
return 0;
}