划分技能点相等的团队

204 阅读3分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 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 <= 105
  • skill.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;
};

说在后面

本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。