华为OD统一考试 - 最小矩阵宽度

题目描述

给定一个矩阵，包含 N * M 个整数，和一个包含 K 个整数的数组。

现在要求在这个矩阵中找一个宽度最小的子矩阵，要求子矩阵包含数组中所有的整数。

输入描述

第一行输入两个正整数 N，M，表示矩阵大小。

接下来 N 行 M 列表示矩阵内容。

下一行包含一个正整数 K。

下一行包含 K 个整数，表示所需包含的数组，K 个整数可能存在重复数字。

所有输入数据小于1000。

输出描述

输出包含一个整数，表示满足要求子矩阵的最小宽度，若找不到，输出-1。

用例

题目解析

本题其实就是“最小覆盖子串”问题的变形。关于最小覆盖子串问题，

当了解了最小覆盖子串问题后，本题中各个关键词可以类比到最小覆盖子串问题中的关键词：

因此，本题可以使用尺取法高效地求出满足要求地子矩阵地最小宽度。

import Foundation

func ODTest_2_19() {
    print("输入描述")
    print("第一行输入两个正整数 N，M，表示矩阵大小。")
    print("接下来 N 行 M 列表示矩阵内容。")
    // 矩阵 [行数, 列数]
    let NM = (readLine() ?? "").split(separator: " ").map { Int($0) ?? 0 }
    if NM.count != 2 {
        print("无")
        return
    }
    let N = NM[0], M = NM[1]
    // 矩阵
    var matrix = Array(repeating: Array(repeating: 0, count: M), count: N)
    for i in 0 ..< N {
        matrix[i] = (readLine() ?? "").split(separator: " ").map { Int($0) ?? 0 }
    }

    print("下一行包含一个正整数 K。")
    // 目标数组长度
    let K = Int(readLine() ?? "") ?? 0
    print("下一行包含 K 个整数，表示所需包含的数组，K 个整数可能存在重复数字。")
    print("所有输入数据小于1000。")
    // 目标数组
    let nums = (readLine() ?? "").split(separator: " ").map { Int($0) ?? 0 }

    // cnts[num] 记录的是 目标数组中num元素的个数
    var cnts = Array(repeating: 0, count: 1000)
    for num i