小Hi和小Ho的礼物

题目：小Hi和小Ho的礼物

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

---

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

分析过程下次补上

代码如下：

 #include <iostream>

 #include <unordered_map>

 using namespace std;

 int main()

 {

     int n, a[];

     long long ans = ;

     unordered_map<int, int> cnt1, cnt2;

     cin >> n;

     //预处理cnt1,cnt1[X]表示“包含X枚金币的袋子个数”

     for(int i = ; i < n; i ++)

     {

         cin >> a[i];

         cnt1[a[i]]++;

     }

     //预处理cnt2，cnt2[X]表示“选出2个袋子，金币之和为X的选法种数”

     for(int i = ; i < n; i ++)

         for(int j = i + ; j < n; j ++)

             cnt2[a[i] + a[j]]++;

     for(int i = ; i < n; i ++)

         for(int j = i + ; j < n; j ++)

         {

             if(a[i] != a[j])    //容斥原理

                 ans += cnt2[a[i] + a[j]] - cnt1[a[i]] - cnt1[a[j]] + ;

             else

                 ans += cnt2[a[i] + a[j]] - cnt1[a[i]] - cnt1[a[j]] + ;

         }

     cout << ans << endl;

     return ;

 } 

运行结果如下：