字节青训营入门算法题：飞行棋分组
链接：飞行棋分组🔗🔗
题目
现在有一堆飞行棋棋子，每个棋子上标有数字序号。需要将这些棋子分成若干组，每组包含5个棋子，且组内所有棋子的数字序号必须相同。需要判断是否可以完成这样的分组。
解答
为了解决这个问题，我们可以使用以下步骤：
- 对棋子序号进行排序。
- 遍历排序后的棋子序号，检查每组连续相同序号的棋子数量是否为5。
- 如果在遍历过程中发现任何一组棋子数量不等于5，则返回
false。 - 如果所有组都满足条件，则返回
true。
知识点
- 数组排序：使用
sort()方法对数组进行排序。 - 遍历数组：使用
for循环遍历数组元素。 - 条件判断：使用
if语句来检查条件是否满足。
讲解
以下是代码实现，其中包含了对之前实现的修正：
function solution(nums) {// 将输入字符串转换为数字数组let pieces = nums.split(' ').map(Number);// 对棋子序号进行排序pieces.sort((a, b) => a - b);// 初始化起始和结束指针let start = 0;let end = 0;// 遍历排序后的棋子序号while (end < pieces.length) {// 移动结束指针直到遇到不同的棋子序号while (end < pieces.length && pieces[end] === pieces[start]) {end++;}// 检查当前组棋子数量是否为5if ((end - start) !== 5) {return false; // 如果不是5，则无法完成分组，返回false}// 更新起始指针到下一组start = end;}// 如果所有组都满足条件，返回truereturn true;
}
// 测试样例
console.log(solution("1 2 3 4 5")); // 输出 false
console.log(solution("1 1 1 1 2 1 2 2 2 2")); // 输出 true
这段代码首先将输入字符串转换为数字数组，并对数组进行排序。然后使用两个指针 start 和 end 来遍历数组，检查每组连续相同序号的棋子数量是否为5。如果在任何点上不满足这个条件，函数会立即返回 false。如果所有组都满足条件，函数在结束时返回 true。
在代码中，我修正了几个问题：
- 输入应该是字符串，需要先转换为数组。
sort()方法默认按照字符串排序，所以需要传递一个比较函数来正确排序数字。end的初始化应该为0，因为我们从数组的第一个元素开始。- 最后，返回值应该是布尔类型，而不是字符串。