hihocoder 1505

hihocoder 1505

题意：给你n个数，让你从n个数中抽两个数，再抽两个数，使得前两个数和后两个数相等

分析：对 i,j,p,q遍历的话时间复杂度会达到o(n4),所以考虑优化p，q

　　

假设分配给小hi的金币为 a[i]和a[j]，现在的问题是求剩下的元素中有多少组 p, q 使得 a[i] + a[j] = a[p] + a[q]。如果暴力枚举所有可能的 p, q，需要 O(n^2)时间，再加上枚举i, j 的时间，时间复杂度是 O(n^4)，会超时。所以不能枚举 p, q。

把输入保存在数组 a[] 中。

令 sumCnt[x] 表示两袋金币之和 a[i] + a[j] = x 组合数。

令 cnt[y] 表示元素 y 出现的次数。

假设分配给小hi的金币为 x = a[i] + a[j]，那么x一共有 sumCnt[x] 种可能的组合。

令 c1 = cnt[a[i]], c2 = cnt[a[j]]，

假设 a[i] != a[j]。

去掉a[i], a[j]之前，a[i]和a[j]一共可以组成 c1 * c2个 x。去掉 a[i] 后，还有 c1 - 1个与 a[i] 相同的元素，去掉 a[j] 后，还有 c2 - 1 个与 a[j] 相同的元素，它们还可以组成 (c1 - 1) * (c2 - 1) 个 x。

所以，如果分配给小hi的金币为 a[i]和a[j]，那么存在 sumCnt - (c1 * c2 - (c1 - 1) * (c2 - 1)) 对 p, q 使得 a[i] + a[j] = a[p] + a[q]。

当 a[i] == a[j] 时，也是类似的处理。

现在，求p, q的组数只需要 O(1)，总的时间复杂度是 O(n^2)。

如果不考虑非法组合的话， 辣么 对于 a[i] + a[j] = a[q] + a[p] = M。 假如 有x 对 a[i] + a[j] = M 的话，答案就是 C(m,2).
考虑重复。 举个栗子， 对于序列， 1 1 2 2 2。 当你M = 3,选择（1,3)(下标)时，你再选择其他（a[q],a[p]）的情况，你要去掉 下标（1,4,）（1,5）（3,2）。 也就是 包含（1，3）中某一个的情况。可以发现有 （n+m-2）n为1个数，m为3个数。

1：v[

2；sumcnt[v[i]+v[j]]函数是存储sum=v[i]+v[j]的个数

 #include <iostream>
 #include <cmath>
 #include <cstdio>
 #include <cstring>
 #include <string>
 #include <map>
 #include <iomanip>
 #include <algorithm>
 #include <queue>
 #include <stack>
 #include <set>
 #include <vector>
 //const int maxn = 1e5+5;
 #define ll long long
 #define MAX INT_MAX
 #define FOR(i,a,b) for( int i = a;i <= b;++i)
 using namespace std;
 int v[],cnt[],sumcnt[];
 ll n,ans;
 int main()
 {
        //    freopen("D:\\common_text\\code_stream\\in.txt","r",stdin);
     //    freopen("D:\\common_text\\code_stream\\out.txt","w",stdout);
     cin>>n;
     FOR(i,,n)
     {
         cin>>v[i];
         cnt[v[i]]++;
     }
     for(int i=;i<=n-;++i)
     {
         for(int j=i+;j<=n;++j)
         {
             sumcnt[v[i]+v[j]]++;
         }
     }
     for(int i=;i<=n-;++i)
     {
         for(int j=i+;j<=n;++j)
         {
             if(v[i]!=v[j])
             {
                 ans+=sumcnt[v[i]+v[j]]-cnt[v[i]]-cnt[v[j]]+1;　　　　
44             }
             else
             {
                 ans+=sumcnt[v[i]+v[j]]-(cnt[v[i]]-1)-(cnt[v[j]]-1)+1;　　　　//这两个看不懂很正常，一定要在纸上自己画画，重叠部分+1，想法也可以不唯一，化简后还是一样的
             }
         }
     }
     cout<<ans<<endl;
 }