[小米OJ] 4. 最长连续数列
思路:
时间限制为O(n),即不能使用先排序后寻找的方法。
这里利用哈希表查询插入复杂度都为O(1)的特性来解,利用一个哈希表来保存每一个数字以及其所在数列的长度。
遍历每一个数字n:查询表中是否存在n-1和n+1,若存在,则hash[n]的值为1 + hash[n-1] + hash[n+1],若不存在即是 1.
同时,为了预防相同数字多次判断,每一次遍历时判断该数字在表中的value是否为0,若不为0说明已经判断过了,跳过即可。
#include <bits/stdc++.h>
using namespace std;
int main()
{
string input;
while (cin >> input)
{
istringstream iss(input);
string temp;
vector<int> vec;
map<int, int> mapping;
int maxNum = -1;
while (getline(iss, temp, ','))
{
vec.push_back(atoi(temp.c_str()));
mapping[atoi(temp.c_str())] = 0;
}
for (auto c : vec)
{
if (mapping[c] == 0)
{
mapping[c] = 1;
if (mapping.count(c - 1))
{
mapping[c] += mapping[c - 1];
}
if (mapping.count(c + 1))
{
mapping[c] += mapping[c + 1];
}
maxNum = maxNum > mapping[c] ? maxNum : mapping[c];
}
}
cout << maxNum << endl;
}
return 0;
}