Longest Square Streak in an Array

export function longestSquareStreak(nums: number[]): number {
  nums.sort((a, b) => a - b);
 
  let answer = -1;
  const map = new Map<number, number>();
  for (const num of nums) {
    const root = Math.sqrt(num);
    if (Number.isInteger(root) && map.has(root)) {
      const rootLength = map.get(root)!;
      map.set(num, rootLength + 1);
      answer = Math.max(answer, rootLength + 1);
    } else {
      map.set(num, 1);
    }
  }
  return answer;
}

export function longestSquareStreak(nums: number[]): number {
  const MAX_NUM = 10 ** 5;
  const set = new Set(nums);
 
  let answer = 0;
  for (const num of nums) {
    let current = num;
    let maxLength = 0;
    while (set.has(current) && current <= MAX_NUM) {
      maxLength += 1;
      current *= current;
    }
    answer = Math.max(answer, maxLength);
  }
  return answer < 2 ? -1 : answer;
}