猫咪特征提取
题目描述
小明是一名算法工程师,同时也是一名铲屎官。某天,他突发奇想,想从猫咪的视频里挖掘一些猫咪的运动信息。为了提取运动信息,他需要从视频的每一帧提取“猫咪特征”。一个猫咪特征是一个两维的vector<x, y>。如果x_1=x_2 and y_1=y_2,那么这俩是同一个特征。
因此,如果喵咪特征连续一致,可以认为喵咪在运动。也就是说,如果特征<a, b>在持续帧里出现,那么它将构成特征运动。比如,特征<a, b>在第2/3/4/7/8帧出现,那么该特征将形成两个特征运动2-3-4 和7-8。 现在,给定每一帧的特征,特征的数量可能不一样。小明期望能找到最长的特征运动。
输入描述:
第一行包含一个正整数N,代表测试用例的个数。
每个测试用例的第一行包含一个正整数M,代表视频的帧数。
接下来的M行,每行代表一帧。其中,第一个数字是该帧的特征个数,接下来的数字是在特征的取值;比如样例输入第三行里,2代表该帧有两个猫咪特征,<1,1>和<2,2>
所有用例的输入特征总数和<100000
N满足1≤N≤100000,M满足1≤M≤10000,一帧的特征个数满足 ≤ 10000。
特征取值均为非负整数。
输出描述:
对每一个测试用例,输出特征运动的长度作为一行
示例1
输入
1
8
2 1 1 2 2
2 1 1 1 4
2 1 1 2 2
2 2 2 1 4
0
0
1 1 1
1 1 1
输出
3
字节跳动的题目大都是读取臭长臭长的输入,所以对输入的读取其实都要花一些时间,不然很容易出错。
思路:
用哈希表顺序存储每一个特征,按照每一行的顺序读取。由于我们需要的是最大连续特征,那么为了保证连续,在读取本行结束时,遍历哈希表内的键,判断是否在本行中。在本行的话,不做处理,不在本行的话,重置其键值为1。
接着记录当前最大键值即可。
import sys
def solution(x):
mp = {}
max_res = 0
for f in x:
for feat in f:
if feat in mp:
mp[feat] += 1
else:
mp[feat] = 1
if mp[feat] > max_res:
max_res = mp[feat]
for feat in mp.keys(): #重置那些不连续的特征为1
if feat not in f:
mp[feat] = 1
return max_res
N = int(sys.stdin.readline().strip())
all_input = []
for i in range(N):
n_frames = int(sys.stdin.readline().strip())
one_input = []
for j in range(n_frames):
one_frame = []
tmp = list(map(int, sys.stdin.readline().strip().split()))
for k in range(tmp[0]):
one_frame.append((tmp[2*k+1], tmp[2*(k+1)]))
one_input.append(one_frame)
all_input.append(one_input)
for one_input in all_input:
print(solution(one_input))
由于是在线编程,题目通过就算了。其实可以用class方法写得更美观。偷个懒哈!