工作中算法题请教,二维数组计算

2017-10-10 11:45:14 +08:00
 mahone3297

算法题 大家好,遇到一个工作中的问题,类似于算法题,就当算法题来向大家请教

一个数组 m*n(即 m 行,n 列),值只包含 0,1
[0,1,1,1,0,0.....]
[1,1,0,1,1,0.....]
[0,1,0,1,0,1.....]
...
[1,1,0,1,1,0.....]
给予参数 lines, columns
找出 x 行数据(x 必须大于 lines),使得行中的数据,值为 1 的列,总列数不多于 columns

例子 1
input:
[0,1,1,0,0,0]
[0,1,0,1,0,0]
[0,1,1,1,0,0]
[1,1,0,0,1,1]
lines = 2, columns = 3

output:
[0,1,1,0,0,0]
[0,1,0,1,0,0]
即第 1,2,3 行数据
3779 次点击
所在节点    算法
20 条回复
rrfeng
2017-10-10 12:03:42 +08:00
按行遍历过去不就行了?也不麻烦啊,求和之后和 columns 比较即可。

更有效率的算法暂时没想到。

感觉跟二维数组没关系。
mahone3297
2017-10-10 12:11:13 +08:00
@rrfeng 你可能没理解我的意思,可能我表达的不清楚

找出 x 行数据(x 必须大于 lines),使得行中的数据,值为 1 的列, [总列数不多于 columns]
这里的总列数的意思是,所有行加起来的总列数。如:
<pre>
[0,1,1,0,0,0]
[0,0,0,0,1,1]
</pre>
这样的数据,所有行数的总列数加起来 = 4
mahone3297
2017-10-10 12:13:22 +08:00
@rrfeng
[0,1,1,0,0,0]
[0,0,1,0,1,0]
这样的数据,所有行数的总列数加起来 = 3
LuckCode
2017-10-10 12:28:26 +08:00
leetcode 有个类似的好像,建立一个 byte[],遍历一行,对应列置 1,之后检查 1 的个数,不够则下一行重复上述操作,如果只是找出一个满足的结果而不是所有情况的话,应该就可以了吧。
hand515
2017-10-10 13:06:25 +08:00
找出 x 行数据(x 必须大于 lines)

这里的 x 是大于等于 lines 吧?你给的 output 结果 x 不是等于 2 吗?
tomatoz
2017-10-10 13:20:17 +08:00
你说的总列数是想找到多行在一维上的"投影",是吗
,创造一个全是 0 的行,假如就叫 new_line 吧,遍历数组,每行都做这样的操作: 每个元素和 new_line 对应元素按位取或,生成的结果覆盖 new_line,然后对 new_line 里"1"的个数计数。至于 x 什么的你自己应该会判断吧。
RecursiveG
2017-10-10 13:40:49 +08:00
1. 选中所有行
2. 统计选中行组成的矩阵中,每一列有多少“ 1 ”。选择“ 1 ”最少的一列,将这一列为“ 1 ”的行取消选择。
3. 重复步骤 2 直到含“ 1 ”列数量满足 columns 要求
4. 检查选中行数量是否满足 lines 要求,满足则有解,不满足则无解
mahone3297
2017-10-10 13:55:48 +08:00
@hand515 嗯,大于等于,我的描述不正确
@tomatoz 嗯,你说的对。但你说的这个 new_line,解决的只是其中的一步(即找出所有行的总列数),这个题最后的结果,不是找列,是需要返回行
mahone3297
2017-10-10 14:03:16 +08:00
@RecursiveG
* 非常感谢您的答案!你的答案,确实给出了一个解。但,好像不是最优解感觉。
* 你的算法,和下面的算法类似(下面的算法是我目前的算法),你看看,是不是
* 将所有行,按列,值相加,得到一个数组,值为列出现的次数。比如
input:
[0,1,1,0,0,0]
[0,1,0,1,0,0]
[0,1,1,1,0,0]
[1,1,0,0,1,1]
得到数组[1,4,2,2,1,1]
* 从大到小的顺序,取出前 columns 列,假如 columns=3,则前 3 位为 4,2,2,分别是第 2,3,4 列
* 遍历原二维数组,看每行为 1 的值是否只在上面的 2,3,4 列,得到新二维数组
* 判断新二维数组是否满足 lines 的要求,满足则是解,不满足则无解
minami
2017-10-10 14:13:44 +08:00
如果 n 不大的话,由于每行数据其实是一个非负整数的二进制表示,则原矩阵可以按行转化为 m 维的向量。利用按位与的性质,可以用 DP 解决。
求二进制数中 1 的个数有快速算法,甚至可以用 CPU 指令解决,见
https://www.cnblogs.com/graphics/archive/2010/06/21/1752421.html。
minami
2017-10-10 14:14:47 +08:00
@minami #10 修正,是按位或
RecursiveG
2017-10-10 14:19:16 +08:00
不一样,考虑如下情况
[1,1,0,1,0,0]
[1,0,1,0,1,0]
[0,1,1,0,0,1]
columns=3, lines=1

你的算法:
[2,2,2,1,1,1]
取 1,2,3 列,没有行满足要求,输出无解。

我的算法:
全选所有行:[2,2,2,1,1,1]
选择第 4 列,剔除第 1 行,更新各列和:[1,1,2,0,1,1]
选择第 1 列,剔除第 2 行,更新各列和:[0,1,1,0,0,1]
已满足 columns 要求,并且目前仍有第 3 行被选中
满足 lines 要求,输出有解。
算法复杂度目测 O(m(n^2))
rrfeng
2017-10-10 14:25:00 +08:00
哦,我说不可能这么简单...
mkeith
2017-10-10 15:29:15 +08:00
二进制 按位异或 运算呢?
mahone3297
2017-10-10 16:32:00 +08:00
@minami 我不是要求 1 的个数。我期望的 output,是行
@RecursiveG 确实,好像你的算法更优!有一些证明吗?你的算法确实比我的好,或者你的算法,找到的就是最优解?你的这个算法,有通用模型吗?你是如何想到的?还是直接遇到过类似的题目?
minami
2017-10-10 18:31:23 +08:00
@mahone3297 你没懂我的意思,你 dp 后就可以取出满足要求的所有区间。求 1 的个数是为了求你所谓的总列数,这是等价问题。
RecursiveG
2017-10-11 04:04:45 +08:00
@mahone3297
好吧我的方法也是错的。反例:
[0,1,1,1,0,0,0]
[1,0,1,1,0,0,0]
[1,1,0,1,0,0,0]
[1,1,1,0,0,0,0]
[0,0,0,0,1,1,1]
[0,0,0,0,1,1,1]
lines=2
columns=3

@minami
求 DP 方程看看?
mahone3297
2017-10-11 09:51:31 +08:00
@RecursiveG 嗯,我昨天发完贴,又想,你和我,某种程度上是一样的。我从大往小,你从小往大。某种程度上,都会造成,a 适用,b 不适用的问题

@minami 嗯,麻烦再多说点,给 DP 方程看看
minami
2017-10-11 23:59:00 +08:00
@mahone3297 #18 你要找的 x 行数据是连续的还是不连续的?而且我看 12L 和 17L 的讨论又糊涂了,RecursiveG 举得两个例子不是都无解么
minami
2017-10-12 01:36:49 +08:00

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/396416

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX