牛客网 字节跳动算法题-特征提取

描述

小明是一名算法工程师,同时也是一名铲屎官。某天,他突发奇想,想从猫咪的视频里挖掘一些猫咪的运动信息。为了提取运动信息,他需要从视频的每一帧提取“猫咪特征”。一个猫咪特征是一个两维的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);
        }
    }
}
经验分享 程序员 微信小程序 职场和发展