小Hi和小Ho的礼物

 

题目：小Hi和小Ho的礼物

注：【i、j、p、q】为下标

---

 

 个人感觉这道题是有一定难度的。读者可以参考一下【四平方和】的解题思路

 

分析过程下次补上

 

代码如下：

#include <iostream>
#include <unordered_map>
using namespace std;

int main()
{
    int n, a[1000];
    long long ans = 0;
    unordered_map<int, int> cnt1, cnt2;
    
    cin >> n;
    
    //预处理cnt1,cnt1[X]表示“包含X枚金币的袋子个数”
    for(int i = 0; i < n; i ++)
    {
        cin >> a[i];
        cnt1[a[i]]++; 
    }
    //预处理cnt2，cnt2[X]表示“选出2个袋子，金币之和为X的选法种数”
    for(int i = 0; i < n; i ++)
        for(int j = i + 1; j < n; j ++)
            cnt2[a[i] + a[j]]++;
    
    for(int i = 0; i < n; i ++)
        for(int j = i + 1; j < n; j ++)
        {
            if(a[i] != a[j])    //容斥原理 
                ans += cnt2[a[i] + a[j]] - cnt1[a[i]] - cnt1[a[j]] + 1;
            else
                ans += cnt2[a[i] + a[j]] - cnt1[a[i]] - cnt1[a[j]] + 3;
        }
    cout << ans << endl;
     
    return 0;
} 

 

运行结果如下：