OD C卷 - 宽度最小的子矩阵

宽度最小的子矩阵 (100)

  • 给定一个n行 * m列的矩阵;
  • 给定一个k个整数的数组k_list;
  • 在n*m的矩阵中找一个宽度最小的子矩阵,该子矩阵包含k_list中所有的整数;
    输入描述:
    第一行输入n,m 两个整数;
    后续n行每行输入 m个数据;
    输入k值;
    输入个整数
    输出描述:
    最小宽度值,若找不到,则输出-1

示例1

输入:

2 5

1 2 2 3 1

2 3 2 3 2

3

1 2 3

输出:

2

说明,

矩阵第0、3列包含了1、2、3;

矩阵第3、4列包含了1、2、3

示例2

输入:

2 5

1 2 2 3 1

1 3 2 3 4

3

1 1 4

输出:

5
思路:

  • 滑动的子矩阵
  • 从第一列起始,找一个宽度最小的子矩阵;
  • 从第二列开始,找一个宽度最小的子矩阵;
  • 依次到最后一列...
  • 以上的宽度每次取最小值
python 复制代码
 
class MinWidth:
    def solution(self, n, m, matrix, k_list):
        k_dict = self.to_dict(k_list)
        min_width = float("inf")
        # 类似双指针
        for start_idx in range(m):
            for end_idx in range(start_idx, m):
                temp_list = []
                # 获取当前子矩阵的所有元素
                for i in range(n):
                    temp_list.extend(matrix[i][start_idx:end_idx+1])
                temp_dict = self.to_dict(temp_list)
                # 集合操作
                flag = True
                for key in k_dict:
                    if key in temp_dict and k_dict[key] <= temp_dict[key]:
                        continue
                    else:
                        flag = False
                        break

                if flag:
                    min_width = min(min_width, end_idx - start_idx + 1)
                    break
        print(min_width)

    def to_dict(self, alist):
        dict_ = {}
        for i in alist:
            dict_[i] = dict_.get(i, 0) + 1
        return dict_


if __name__ == '__main__':
    min_width = MinWidth()

    while True:
        try:
            n, m = list(map(int, input().strip().split()))
            matrix = []
            for i in range(n):
                matrix.append(list(map(int, input().strip().split())))
            k = int(input().strip())
            k_list = list(map(int, input().strip().split()))
            min_width.solution(n, m, matrix, k_list)

        except KeyboardInterrupt:
            break

&nbsp

相关推荐
f狐0狸x5 天前
【蓝桥杯每日一题】3.28
c语言·数据结构·c++·蓝桥杯·滑动窗口
ゞ 正在缓冲99%…8 天前
leetcode3.无重复字符的最长字串
算法·leetcode·滑动窗口
小卡皮巴拉15 天前
【力扣刷题实战】无重复的最长字串
开发语言·c++·算法·leetcode·滑动窗口
daily_233316 天前
coding ability 展开第四幕(滑动指针——巩固篇)超详细!!!!
c++·算法·字符串·哈希算法·滑动窗口
daily_233320 天前
coding ability 展开第三幕(滑动指针——基础篇)超详细!!!!
android·开发语言·数据结构·c++·算法·滑动窗口
闻缺陷则喜何志丹1 个月前
【二分查找、滑动窗口】P10389 [蓝桥杯 2024 省 A] 成绩统计|普及+
c++·算法·蓝桥杯·二分查找·滑动窗口·洛谷·成绩
WenGyyyL1 个月前
蓝桥杯真题——近似gcd(很离谱!!!)
算法·leetcode·蓝桥杯·滑动窗口
源代码•宸2 个月前
Leetcode—487. 最大连续1的个数 II【中等】Plus
c++·经验分享·算法·leetcode·滑动窗口
神秘的t2 个月前
优选算法合集————双指针(专题二)
java·数据结构·算法·滑动窗口
L_M_TY2 个月前
G1. Yunli‘s Subarray Queries (easy version)
算法·stl·滑动窗口·离线查询