平衡等大小的数组
2013-5-24 15:0:38
收藏:0
阅读:130
评论:2
我不知道“平衡”的正确名称是什么。
我有一个2n个正整数元素的数组,我想将其拆分为两个n个元素的数组,且它们的平均值之间的差异最小。例如:
values: {4, 4, 7, 8, 10, 15}
(some magic here)
a: {7, 8, 10}
b: {4, 4, 15}
我不确定总是将最小数和最大数相结合是否总是会拆分出最少的不同平均值。有没有办法实现这个算法以始终正确地拆分?
点赞
用户1847592
local function some_magic(V)
assert(#V % 2 == 0 and #V > 0)
table.sort(V)
local S = {[-1] = math.huge, [0] = 0}
for i = 1, #V do
S[i] = S[i-1] + V[i]
end
local half_sum = math.floor(S[#V] / 2)
local half_len = #V / 2
local m = 2^math.ceil(math.log(half_len+1)/math.log(2))
local P = {[0] = 0}
for idx = #V, 1, -1 do
local v, P2 = V[idx], {}
for k in pairs(P) do
if math.floor(k/m) + v + S[half_len - k%m - 1] <= half_sum then
P2[k + v*m + 1] = idx
end
end
for k, v in pairs(P2) do
P[k] = P[k] or v
end
end
local k = 0
for next_k in pairs(P) do
if next_k > k and next_k%m == half_len then
k = next_k
end
end
local A, B, prev_idx = {}, {}, 0
repeat
local idx = P[k]
for i = prev_idx + 1, idx - 1 do
table.insert(B, V[i])
end
table.insert(A, V[idx])
prev_idx = idx
k = k - V[idx]*m - 1
until k == 0
for i = prev_idx + 1, #V do
table.insert(B, V[i])
end
return A, B
end
local values = {4, 4, 7, 8, 10, 15}
print('values: '..table.concat(values, ','))
local a, b = some_magic(values)
print('a: '..table.concat(a, ','))
print('b: '..table.concat(b, ','))
------------- Output: ------------
values: 4,4,7,8,10,15
a: 4,4,15
b: 7,8,10
local function some_magic(V)
assert(#V % 2 == 0 and #V > 0) -- 确保输入是偶数个正整数
table.sort(V) -- 排序
local S = {[-1] = math.huge, [0] = 0} -- S 列表用于保存 V 的前缀和
for i = 1, #V do
S[i] = S[i-1] + V[i]
end
local half_sum = math.floor(S[#V] / 2) -- half_sum 是整个列表的一半之和
local half_len = #V / 2 -- half_len 是整个列表的一半长度
local m = 2^math.ceil(math.log(half_len+1)/math.log(2)) -- m 是一个 2 的幂,大于等于 half_len+1
local P = {[0] = 0} -- P 表示一个从 S 中选出的整数子集
for idx = #V, 1, -1 do
local v, P2 = V[idx], {}
for k in pairs(P) do
if math.floor(k/m) + v + S[half_len - k%m - 1] <= half_sum then -- 迭代更新 P
P2[k + v*m + 1] = idx
end
end
for k, v in pairs(P2) do
P[k] = P[k] or v
end
end
local k = 0
for next_k in pairs(P) do -- 在 P 中找到恰好等于 half_len 的 k
if next_k > k and next_k%m == half_len then
k = next_k
end
end
local A, B, prev_idx = {}, {}, 0
repeat
local idx = P[k]
for i = prev_idx + 1, idx - 1 do -- 将这个中间值的左边部分填充到列表 A 中,右边部分填充到 B 中
table.insert(B, V[i])
end
table.insert(A, V[idx])
prev_idx = idx
k = k - V[idx]*m - 1
until k == 0
for i = prev_idx + 1, #V do
table.insert(B, V[i])
end
return A, B
end
local values = {4, 4, 7, 8, 10, 15} -- 原始列表
print('values: '..table.concat(values, ','))
local a, b = some_magic(values) -- 把原始列表分为两个部分
print('a: '..table.concat(a, ','))
print('b: '..table.concat(b, ','))
------------- 输出: ------------
values: 4,4,7,8,10,15
a: 4,4,15
b: 7,8,10
2013-05-24 10:20:21
评论区的留言会收到邮件通知哦~
推荐文章
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1:1导入真实世界的地形?
- 求解,lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教:如何用lua实现游戏内调用数据库函数实现账号密码注册?
- Lua实现网站后台开发
- LUA错误显式返回,社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接,并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下,有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值(不考虑键的顺序)
- 有个lua简单的项目,外包,有意者加微信 liuheng600456详谈,最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码?
- addEventListener 返回 nil Lua
- Lua中获取用户配置主目录的跨平台方法
如果我理解您的需求正确,一种快速而近似正确的解决方案是:
关键是对原始数组进行排序,如果您的要求允许,在地方进行排序;否则,创建一个新的“原始”数组,其中包含实际原始数组的排序副本。
如果 N 相当大,且数字的分布相当均匀(例如没有向一端或另一端倾斜的大偏差),那么此解决方案将非常接近理想解决方案。
如果您真的需要平均值尽可能接近,即使数字存在较大偏差,我能想到的唯一方法是尝试每个排列并保留平均差最小的排列。