二进制矩阵 grid选出好子集下标

题目:
给你一个下标从 0 开始大小为 m x n 的二进制矩阵 grid 。

从原矩阵中选出若干行构成一个行的 非空 子集,如果子集中任何一列的和至多为子集大小的一半,那么我们称这个子集是 好子集。

更正式的,如果选出来的行子集大小(即行的数量)为 k,那么每一列的和至多为 floor(k / 2) 。

请你返回一个整数数组,它包含好子集的行下标,请你将子集中的元素 升序 返回。

如果有多个好子集,你可以返回任意一个。如果没有好子集,请你返回一个空数组。

一个矩阵 grid 的行 子集 ,是删除 grid 中某些(也可能不删除)行后,剩余行构成的元素集合。

示例 1:

输入:grid = [[0,1,1,0],[0,0,0,1],[1,1,1,1]]
输出:[0,1]
解释:我们可以选择第 0 和第 1 行构成一个好子集。
选出来的子集大小为 2 。

第 0 列的和为 0 + 0 = 0 ,小于等于子集大小的一半。
第 1 列的和为 1 + 0 = 1 ,小于等于子集大小的一半。
第 2 列的和为 1 + 0 = 1 ,小于等于子集大小的一半。
第 3 列的和为 0 + 1 = 1 ,小于等于子集大小的一半。
示例 2:
输入:grid = [[0]]
输出:[0]
解释:我们可以选择第 0 行构成一个好子集。
选出来的子集大小为 1 。

第 0 列的和为 0 ,小于等于子集大小的一半。
示例 3:
输入:grid = [[1,1,1],[1,1,1]]
输出:[]
解释:没有办法得到一个好子集。

提示:

m == grid.length
n == grid[i].length
1 <= m <= 10^4
1 <= n <= 5
grid[i][j] 要么是 0 ,要么是 1 。

解答:
public static List goodSubsetofBinaryMatrix(int[][] grid) {
HashMap<Integer, Integer> index = new HashMap<>();
for (int i = 0; i < grid.length; i++) {
int mask = 0;
for (int j = 0; j < grid[i].length; j++) {
mask |= grid[i][j] << j;
}
index.put(mask, i);
}
if (index.containsKey(0)) {
return Arrays.asList(index.get(0));
}
Iterator<Map.Entry<Integer, Integer>> iterator1 = index.entrySet().iterator();
while (iterator1.hasNext()) {
Map.Entry<Integer, Integer> next1 = iterator1.next();
Iterator<Map.Entry<Integer, Integer>> iterator2 = index.entrySet().iterator();
while (iterator2.hasNext()) {
Map.Entry<Integer, Integer> next2 = iterator2.next();

          if ((next1.getKey() & next2.getKey()) == 0) {
              int mm = next1.getValue();
              int nn = next2.getValue();
              return mm < nn ? Arrays.asList(mm, nn) : Arrays.asList(nn, mm);
          }
      }
  }
  return Arrays.asList();

}

posted @   vello  阅读(7)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 从HTTP原因短语缺失研究HTTP/2和HTTP/3的设计差异
· 三行代码完成国际化适配,妙~啊~
点击右上角即可分享
微信分享提示