第322场周赛第2题-划分技能点相等的团队

给你一个正整数数组 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^2暴力,考虑一个hashmap,一趟O(n)计算平均和avl = sum(skill)/(skill.size()/2),同时对于每个skill[i]将map[skill[i]]++,记录下每个数有多少,第二趟O(n)中若map[i] == map[avl-i]且均不为0,则凑成map[i]对.

class Solution {
    int ans = 0;//平均和
    long long temp = 0;
    long long m[2005] = {0};
public:
    long long dividePlayers(vector<int>& skill) {
        int k = skill.size();
        for(int i = 0;i<k;i++)
        {
            m[skill[i]]++;
            ans+=skill[i];
        }
        if(ans%(k/2)==0)
        ans/=(k/2);
        else
        return -1;
        k/=2;
        for(int i = 1;i<=ans;i++)
        {
            if(m[i]!=0&&m[ans-i]!=0)
            {
                if(m[i]==m[ans-i])
                {
                if(i==ans-i)//若i==ans-i,则凑成m[i]/2对,若skill=[1,1,1,1],则m[1] = 4,但实际上是凑成两对
                {
                    temp+=(long long)(i*(ans-i)*(m[i]/2));
                    k -= m[i]/2;
                }
                
                else
                {
                    temp+=(long long)(i*(ans-i)*m[i]);
                    k -= m[i];
                }
                m[i] = 0;
                m[ans-i] = 0;
                }
                else
                return -1;
            }
        }
        if(!k)//若能凑出k对
        return temp;
        else
        return -1;
    }
};