牛客网编程题:猫咪特征提取

猫咪特征提取

编程题目地址

题目描述

小明是一名算法工程师,同时也是一名铲屎官。某天,他突发奇想,想从猫咪的视频里挖掘一些猫咪的运动信息。为了提取运动信息,他需要从视频的每一帧提取“猫咪特征”。一个猫咪特征是一个两维的vector。如果x_1=x_2 and y_1=y_2,那么这俩是同一个特征。

因此,如果喵咪特征连续一致,可以认为喵咪在运动。也就是说,如果特征在持续帧里出现,那么它将构成特征运动。比如,特征在第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方法写得更美观。偷个懒哈!

你可能感兴趣的:(牛客网题目)