牛客网 字节跳动算法题-特征提取
描述
小明是一名算法工程师,同时也是一名铲屎官。某天,他突发奇想,想从猫咪的视频里挖掘一些猫咪的运动信息。为了提取运动信息,他需要从视频的每一帧提取“猫咪特征”。一个猫咪特征是一个两维的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。 特征取值均为非负整数。
输出描述:
对每一个测试用例,输出特征运动的长度作为一行
题解:
使用翻滚HashMap<String,Integer>key代表特征值,value代表每一个特征值目前的特征长度。用两个HashMap,temp和last来保存当前行和上一行的数据,并通过比较当前行和上一行特征值来更新上一行last的数据,然后进入下一行的遍历。本题考查了HashMap的使用,并且采用了翻滚数组的思想。详情请见代码注释。
import java.util.*; public class Main{ public static void main (String []args){ Scanner scanner = new Scanner(System.in); //获取循环次数N,并且转成整数 int N = Integer.parseInt(scanner.nextLine()); while(N-- > 0) { //获取M行,并且需要转成整数 int M = Integer.parseInt(scanner.nextLine()); //初始化一次循环要的变量 int MAX = 1; //temp记录当前行特征值出现的次数 HashMap<String,Integer> temp = new HashMap<String,Integer>(); //last记录上一行特征值出现的次数 HashMap<String,Integer> last = new HashMap<String,Integer>(); for(int i = 0; i < M; i++) { //获取每一行数据,使用String数组保存 String[] arr = scanner.nextLine().split(" "); //获取每一行特征值个数num int num = Integer.parseInt(arr[0]); //清空当前行的temp temp.clear(); for(int j = 1; j <= 2*num; j = j+2) { String s = arr[j]+,+arr[j+1]; temp.put(s, temp.getOrDefault(s,0)+1); } //当前行temp的特征值与上一行保存的特征值比较相加 Set<String> set = last.keySet(); for(String key : set) { if(temp.containsKey(key)) { temp.put(key,temp.get(key) + last.get(key)); MAX = Math.max(MAX,temp.get(key)); } } //清空上一行last,并且把最新的当前行给last赋值 last.clear(); last.putAll(temp); } System.out.println(MAX); } } }