[编程题]特征提取 字节跳动 Golang

[编程题]特征提取

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

输入
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,1>在连续的帧中连续出现3次，相比其他特征连续出现的次数大，所以输出3

备注:

如没有长度大于2的特征运动，返回1

思路

最近再写Golang，但看到这道题马上想到C++的pair和map。思路很简单，就是用map记录次数，但是最开始写的版本特别啰嗦，而别人的C++代码一入眼竟让人误以为是Python，阅读体验仿佛便秘两周一朝通畅的舒爽。不行，我也要重写一份简洁版。
另外，题目备注里的说 大于2 ，实际上应该是 大于等于2 。有一个用例长度为2，输出也为2。

如没有长度大于2的特征运动，返回1

我的解答

#include <iostream>
#include <map>
#include <vector>

using namespace std;

int main()
{
    int m, n, k;
    cin >> n >> m;
    pair <int, int> p;  //记录特征
    while (n--) {       //n个用例           //k个特征
        int maxCnt = 2;
        map<pair<int, int>, int> lastFrame;    //上一帧计数
        map<pair<int, int>, int> currentFrame; //当前帧计数
        while (m--) {   //m帧视频
            cin >> k;        
            while (k--) {
                cin >> p.first >> p.second;
                if (lastFrame.count(p)) {
                    currentFrame[p] = lastFrame[p] + 1;
                    maxCnt = currentFrame[p] > maxCnt ? currentFrame[p] : maxCnt;
                }else currentFrame[p] = 1;
            }
            lastFrame.clear();
            swap(lastFrame, currentFrame);
        }
        cout << maxCnt << endl;
    }
}