Leetcode 1711. 大餐计数

大餐 是指 恰好包含两道不同餐品 的一餐，其美味程度之和等于 2 的幂。

你可以搭配 任意 两道餐品做一顿大餐。

给你一个整数数组 deliciousness ，其中 deliciousness[i] 是第 i 道餐品的美味程度，返回你可以用数组中的餐品做出的不同 大餐 的数量。结果需要对 109 + 7 取余。

注意，只要餐品下标不同，就可以认为是不同的餐品，即便它们的美味程度相同。

示例 1：

输入：deliciousness = [1,3,5,7,9]
输出：4
解释：大餐的美味程度组合为 (1,3) 、(1,7) 、(3,5) 和 (7,9) 。
它们各自的美味程度之和分别为 4 、8 、8 和 16 ，都是 2 的幂。

示例 2：

输入：deliciousness = [1,1,1,3,3,3,7]
输出：15
解释：大餐的美味程度组合为 3 种 (1,1) ，9 种 (1,3) ，和 3 种 (1,7) 。

提示：

• 1 <= deliciousness.length <= 105
• 0 <= deliciousness[i] <= 220

还是时间超时

Code:

class Solution {
public:
    int countPairs(vector<int>& deliciousness) {
        
        sort(deliciousness.begin(),deliciousness.end());
        uint64_t res=0;
        vector<int>tempvec;
        unordered_map<int,int>mymap;
        pair<unordered_map<int, int>::iterator, bool> ret;
        int count=1;
        for(int i=0;i<deliciousness.size();i++)
        {
            ret=mymap.insert(pair<int,int>(deliciousness[i],1));
            if(!ret.second)
            {
                //            cout<<"err"<<endl;
                mymap[deliciousness[i]]=++count;
            }
            else {
                count=1;
                tempvec.push_back(deliciousness[i]);
            }
        }
        
        /*  unordered_map<int, int>::iterator iter;
    for(iter=mymap.begin();iter!=mymap.end();++iter)
    {
        tempvec.push_back(iter->first);
//        cout<<iter->first<<"  "<<iter->second<<endl;
    }*/
        
        
        for(int i=0;i<tempvec.size();i++)
        {
            if(tempvec[i]==0)
                continue;
            if (0 == (tempvec[i] & (tempvec[i ]- 1))&&(mymap[tempvec[i]]>1))
            {
                res+=((1+mymap[tempvec[i]]-1)*(mymap[tempvec[i]]-1))/2;
            }
        }
        
        //    cout<<res<<"++"<<endl;
        
        
        for(int i=0;i<tempvec.size();i++)
        {
            for(int j=i+1;j<tempvec.size();j++)
            {
                int temp=tempvec[i]+tempvec[j];
                if(temp==0||((temp%2!=0)&&temp!=1))//排除零和奇数
                {
                    continue;
                }
                if (0 == (temp & (temp - 1)))
                {
                    
                    //                cout<<mymap[tempvec[i]]<<"-----"<<mymap[tempvec[j]]<<endl;
                    res+=(mymap[tempvec[i]])*(mymap[tempvec[j]]);
                    //               cout<<tempvec[i]<<" "<<tempvec[j]<<endl;
                    //                res++;
                    
                }
                
            }
        }
        
        return res%1000000007;
    } 
};

官方题解：

class Solution {
public:
    static constexpr int MOD = 1000000007;
    int countPairs(vector<int>& deliciousness) {
        int maxVal = *max_element(deliciousness.begin(), deliciousness.end());
        int maxSum = maxVal * 2;
        int pairs = 0;
        unordered_map<int, int> mp;
        int n = deliciousness.size();
        for (auto& val : deliciousness) {
            for (int sum = 1; sum <= maxSum; sum <<= 1) {
                int count = mp.count(sum - val) ? mp[sum - val] : 0;
                pairs = (pairs + count) % MOD;
            }
            mp[val]++;
        }
        return pairs;
    }
};