6254. 划分技能点相等的团队

开启掘金成长之旅！这是我参与「掘金日新计划 · 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 <= 10^5
• 1 <= skill[i] <= 1000

解题思路

满足条件的数组长度必定是偶数，每个团队的总分一样；

1. 求出数组的总分，和团队的个数，求出每个团队的分数avg；
2. 将数组中的元素都放到map中，value为元素出现的次数。
3. 遍历数组元素，看是否在map中有当前数+对应的数=avg的，也就是在map中找avg-当前数，如果没有直接返回-1；
4. 全部完成返回各个团队相乘的结果。

代码

class Solution {
    public long dividePlayers(int[] skill) {
        int n = skill.length;
        if ((n & 1) == 1) return -1;
        int sum = Arrays.stream(skill).sum();
        int avg = sum/(n >> 1);
        Map<Integer,Integer> map = new HashMap<>();
        for (int num : skill) {
            map.put(num,map.getOrDefault(num,0) + 1);
        }
        long ans = 0;
        for (int num : skill) {
            if (map.get(num) > 0) {
                if (!map.containsKey(avg - num) || map.get(avg - num) < 1) {
                    return -1;
                }
                map.put(num,map.get(num) - 1);
                map.put(avg-num,map.get(avg-num) - 1);
                ans = ans + num * (avg - num);
            }
        }
        return ans;
    }
}