码迷,mamicode.com
首页 > 其他好文 > 详细

HiHo1505:小Hi和小Ho的礼物(Meet-In-The-Middle + 组合数学)

时间:2017-05-10 22:24:53      阅读:316      评论:0      收藏:0      [点我收藏+]

标签:hid   sum   print   height   struct   ++   isp   输入   idt   

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

某人有N袋金币,其中第i袋内金币的数量是Ai。现在他决定选出2袋金币送给小Hi,再选2袋金币送给小Ho,同时使得小Hi和小Ho得到的金币总数相等。他想知道一共有多少种不同的选择方法。

具体来说,有多少种下标四元组(i, j, p, q)满足i, j, p, q两两不同,并且i < j, p < q, Ai + Aj = Ap + Aq。

例如对于数组A=[1, 1, 2, 2, 2],一共有12种选法:

i j p q
1 3 2 4
1 3 2 5
1 4 2 3
1 4 2 5
1 5 2 3
1 5 2 4
2 3 1 4
2 3 1 5
2 4 1 3
2 4 1 5
2 5 1 3
2 5 1 4

输入

第一行包含一个整数N。

第二行包含N个整数,A1, A2, A3 ... AN。

对于70%的数据,1 <= N <= 100

对于100%的数据,1 <= N <= 1000, 1 <= Ai <= 1000000

输出

不同选择的数目。

样例输入

5  
1 1 2 2 2

样例输出

12

 技术分享

Code:

技术分享
 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 static const int MAXN = 1e3 + 10;
 4 static const int HS = 2000000 + 10;
 5 typedef long long LL;
 6 struct Node
 7 {
 8     int x , y , num;
 9     Node()
10     {
11         num = 0;
12     }
13 };
14 LL vis[HS];
15 LL dvis[HS];
16 LL data[MAXN];
17 LL ans;
18 int n;
19 int main()
20 {
21     scanf("%d" , &n);
22     for(int i = 1 ; i <= n ; ++i)
23         scanf("%lld" , &data[i]) , ++dvis[data[i]];
24     for(int i = 1 ; i <= n ; ++i)
25     {
26         for(int j = i + 1 ; j <= n ; ++j)
27         {
28             ++vis[data[i] + data[j]];
29         }
30     }
31 
32     for(int i = 1 ; i <= n ; ++i)
33     {
34         for(int j = i + 1 ; j <= n ; ++j)
35         {
36             if(data[i] != data[j])
37             {
38                 LL a = dvis[data[i]] , b = dvis[data[j]];
39                 LL sum = data[i] + data[j];
40                 ans += vis[sum] - (a * b - (a - 1) * (b - 1));
41             }
42             else
43             {
44                 LL sum = data[i] + data[j];
45                 LL pre = dvis[data[i]];
46                 LL last = pre - 2;
47                 ans += vis[sum] - (pre * (pre - 1) - last * (last - 1)) / 2;
48             }
49         }
50     }
51     printf("%lld" , ans);
52 }
View Code

 

HiHo1505:小Hi和小Ho的礼物(Meet-In-The-Middle + 组合数学)

标签:hid   sum   print   height   struct   ++   isp   输入   idt   

原文地址:http://www.cnblogs.com/jianglingxin/p/6838416.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!