校招真题练习017 特征提取(头条)

特征提取

题目描述
小明是一名算法工程师，同时也是一名铲屎官。某天，他突发奇想，想从猫咪的视频里挖掘一些猫咪的运动信息。为了提取运动信息，他需要从视频的每一帧提取“猫咪特征”。一个猫咪特征是一个两维的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。
特征取值均为非负整数。

输出描述:
对每一个测试用例，输出特征运动的长度作为一行

def main():
    #test = [[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]]
    N = int(input())
    #N = 1
    for i in range(N):
        M = int(input())
        #M = 8
        dic = {}
        for j in range(M):
            line = list(map(int,input().split(' ')))
            #line = test[j]
            T = line[0]
            ts = line[1:]
            n = len(ts)
            k = 0
            step = 2
            while k < n // 2:
                key = (ts[step * k],ts[step * k+1])
                if key in dic:
                    dic[key].append(j)
                else:
                    dic[key] = [j]
                k += 1
        maxlen = 0
        for z in dic.values():
            l = len(z)
            if l == 1:
                maxlen = max(maxlen,1)
            else:
                left = 0
                pre = z[0]
                idx = 1
                while idx < l:
                    if z[idx] - pre == 1:
                        pre = z[idx]
                        idx += 1
                    else:
                        maxlen = max(maxlen,idx-left)
                        left = idx
                        idx += 1
                maxlen = max(maxlen,idx-left)
        print(maxlen)

if __name__ == '__main__':
    main()

算法思路：数组遍历，哈希。

建立字典dic，以tuple为键，以该tuple所在的帧数为值(列表)。

遍历字典dic，计算所有的tuple连续帧数的长度，并保留最长的长度，即为所求。