2024华为OD机试真题目录-(B卷C卷D卷)-【C++ Java Python】
题目描述
给定一个矩阵,包含 N * M 个整数,和一个包含 K 个整数的数组。
现在要求在这个矩阵中找一个宽度最小的子矩阵,要求子矩阵包含数组中所有的整数。
输入描述
第一行输入两个正整数 N,M,表示矩阵大小。
接下来 N 行 M 列表示矩阵内容。
下一行包含一个正整数 K。
下一行包含 K 个整数,表示所需包含的数组,K 个整数可能存在重复数字。
所有输入数据小于1000。
输出描述
输出包含一个整数,表示满足要求子矩阵的最小宽度,若找不到,输出-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,
所以最小的矩阵宽度由3,4列组成,为2
考点
滑动窗口
代码
python
n,m=list(map(int,input().split()))
mat=[]
for i in range(n):
row=list(map(int,input().split()))
mat.append(row)
matr=[] #转置后的矩阵
for i in range(m):
tmp=[]
for row in mat:
tmp.append(row[i])
matr.append(tmp)
num=int(input())
arr=list(map(int,input().split()))
arr_set=set(arr)
leng=len(arr_set) # 目标集合中的元素个数
tmps=[] #存列中有用的元素
left=0
res=m #存最小列数
for right in range(m):
for i in matr[right]:
if i in arr_set:
tmps.append(i)
while