字节跳动2019年校招笔试特征提取（Java实现）

题目描述

       小明是一名算法工程师，同时也是一名铲屎官。某天，他突发奇想，想从猫咪的视频里挖掘一些猫咪的运动信息。为了提取运动信息，他需要从视频的每一帧提取“猫咪特征”。一个猫咪特征是一个两维的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 自定义数据特征的结构Array并实现hashCode和equals。

2 定义两个hashmap，分别为lastIndex和Index，其中key的形式为Array，lastIndex中的value值为特征对上次出现的层数，Index中的value值为特征对出现的次数。

3 取出特征对，若上次出现是在上一层，则出现次数+1，否则将出现次数置为1.

4 每输入一层都要找出当前Index中最大的value值并记录。

代码实现：

import java.util.*;
class Array{
    int first,second;
    @Override
    public int hashCode() {
        return first*37+second;
    }
 
    @Override
    public boolean equals(Object obj) {
        return this.first == (((Array)obj).first) && this.second== ((Array)obj).second;
    }
}
public class Main {
        public static void main(String[] args) {
            int i,j = 0,max = 1;
            Scanner sc = new Scanner(System.in);
            int M  = Integer.parseInt(sc.nextLine());  //获取测试用例个数
            HashMap<Array, Integer> lastIndex = new HashMap<>();
            HashMap<Array, Integer> Index = new HashMap<>();
            while(M > 0){
                int N  = Integer.parseInt(sc.nextLine());
                int tag = N;
                while(N > 0){
                    String new_str = "";
                    if(sc.hasNextLine())
                        new_str = sc.nextLine();
                    j++;
                    String str[] = new_str.split(" ");
                    int num = Integer.parseInt(str[0]);
                    if(num ==1 && tag == 1){
                        break;
                    }
                    N--;
                    for(i = 1;i<str.length;i+=2){
                        Array arr = new Array();
                        arr.first = Integer.parseInt(str[i]);
                        arr.second = Integer.parseInt(str[i+1]);
                        boolean b = false;
                        for (Array key:lastIndex.keySet()) {
                            //if(key.first == arr.first && key.second == arr.second && lastIndex.get(key) == j-1){
                            if(key.equals(arr) && lastIndex.get(key) == j-1){
                                lastIndex.put(key,j);
                                b = true;
                                if(Index.size() == 0)
                                    Index.put(key, 2);
                                else{
                                    boolean op = false;
                                    for (Array key2:Index.keySet()) {
                                        if (key2.equals(key)) {
                                            op = true;
                                            Index.put(key2, Index.get(key2) + 1);
                                            break;
                                        }
                                    }
                                    if(op == false)
                                        Index.put(key, 2);
                                }
                                break;
                            }else if (key.first == arr.first && key.second == arr.second && lastIndex.get(key) < j - 1) {
                                lastIndex.put(key,j);
                                b = true;
                                Index.put(key, 1);
                                break;
                            }
                        }
                        if(b == false)
                            lastIndex.put(arr,j);
                        for (Array key : Index.keySet()) {
                            if(Index.get(key) > max)
                                max = Index.get(key);
                        }
                    }
                }               
                System.out.println(max);
                M--;
            }
        }
}