开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第10天,点击查看活动详情
题目描述
给你一个正整数数组 skill ,数组长度为 偶数 n ,其中 skill[i] 表示第 i 个玩家的技能点。将所有玩家分成 n / 2 个 2 人团队,使每一个团队的技能点之和 相等 。
团队的 化学反应 等于团队中玩家的技能点 乘积 。
返回所有团队的 化学反应 之和,如果无法使每个团队的技能点之和相等,则返回 -1 。
示例 1:
输入: skill = [3,2,5,1,3,4]
输出: 22
解释:
将玩家分成 3 个团队 (1, 5), (2, 4), (3, 3) ,每个团队的技能点之和都是 6 。
所有团队的化学反应之和是 1 * 5 + 2 * 4 + 3 * 3 = 5 + 8 + 9 = 22 。
示例 2:
输入: skill = [3,4]
输出: 12
解释:
两个玩家形成一个团队,技能点之和是 7 。
团队的化学反应是 3 * 4 = 12 。
示例 3:
输入: skill = [1,1,2,3]
输出: -1
解释:
无法将玩家分成每个团队技能点都相等的若干个 2 人团队。
提示:
2 <= skill.length <= 105skill.length是偶数1 <= skill[i] <= 1000
思路分析
首先我们要先理解一下题意,题目会给我们一个数组,数组中放有n个玩家的技能点,我们需要将所有玩家两两分组,且每组的技能点之和都要相等,我们需要计算分完组之后的化学反应和,每一组的化学反应为团队中的玩家技能点乘积。
- 1、计算每组团队的技能点和
因为我们是两两分组且每组的技能点和要相等,所以每组的技能点和应该等于总技能点和除于分组数量。
let sum = 0,map = {};
for(let i = 0; i < skill.length; i++){
map[skill[i]] = (map[skill[i]] || 0) + 1;
sum += skill[i];
}
const tn = skill.length / 2;
if(sum % tn != 0) return -1;
sum /= tn;
- 2、判断是否可以所有人都成组
我们需要统计每一个技能点数的玩家数,判断是否有足够的互补玩家进行匹配,如果匹配人数不一致则说明无法所有人都成组,否则技能点和为技能点*互补技能点*人数,这里需要注意下,有可能是两个技能点相同的人成组,这时候人数需要除于2。
for(let key in map){
if(map[sum - key] != map[key]) return -1;
let num = map[key];
if(sum - key == key) num /= 2;
res += (sum - key) * key * num;
delete map[key];
delete map[sum - key];
}
return res;
AC代码
完整AC代码如下:
/**
* @param {number[]} skill
* @return {number}
*/
var dividePlayers = function(skill) {
let sum = 0,map = {};
for(let i = 0; i < skill.length; i++){
map[skill[i]] = (map[skill[i]] || 0) + 1;
sum += skill[i];
}
const tn = skill.length / 2;
if(sum % tn != 0) return -1;
sum /= tn;
let res = 0;
for(let key in map){
if(map[sum - key] != map[key]) return -1;
let num = map[key];
if(sum - key == key) num /= 2;
res += (sum - key) * key * num;
delete map[key];
delete map[sum - key];
}
return res;
};
说在后面
本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。