leetcode:第 224 场周赛:5243. 同积元组(简单,哈希表)

題目:

在这里插入图片描述

分析:

大意啊,
自己还以为是4数相加问题,三重循环+二分/哈希,一直T,最后意识到不是四数之和,直接二重循环找出所有的和,c(n,2)然后就Ok啦!
可惜,要不排名就能再创新高了,哎

代码:

class Solution {
public:
    int tupleSameProduct(vector<int>& nums) {
        map<long long,int> m;
        sort(nums.begin(),nums.end());
        vector<long long> v;
        //int ans=0;
        for(int i=0;i<nums.size()-1;i++)
        {
            for(int j=i+1;j<nums.size();j++)
            {
                m[(long long)nums[i]*(long long)nums[j]]++;
                if(m[(long long)nums[i]*(long long)nums[j]]==1) v.push_back((long long)nums[i]*(long long)nums[j]);
            }
        }
        int ans=0;
        for(int i=0;i<v.size();i++)
        {
            if(m[v[i]]>=2)
            {
                int c=m[v[i]];
                ans+=(long long)c*(c-1)/2;
            }
        }
        return ans*8;
    }
};

你可能感兴趣的:(leetcode:第 224 场周赛:5243. 同积元组(简单,哈希表))