首页 > 代码库 > hdu 5101 Select

hdu 5101 Select

http://acm.hdu.edu.cn/showproblem.php?pid=5101

在比赛的时候没有想出怎么做,自己真水。

题意:给定一些集合,选择两个来自不同集合的数,加和大于k,问有多少种选择方案。

思路:答案=从所有数中选择的两个加和大于k的数的方案数-在同一个集合中选择的两个加和大于k的数的方案数而对于同一个集合中选择的两个加和大于k的方案数是可以直接排序然后利用单调性快速统计出来的。利用vector把所有的数存到q[0]里面。对q[0]排序,也对q[i]排序。

然后在q[0]找每一个k-v+1的位置,用q.end()减去这个位置就是大于等于这个k-v+1的个数,求出这个个数之后再减去在同一个集合里面的个数就是符合要求,这里的v是存入的数。

求出的ans是答案的二倍,ans/2就是答案。

 

 1 #include <cstdio> 2 #include <cstring> 3 #include <vector> 4 #include <algorithm> 5 #define LL __int64 6 using namespace std; 7  8 int t; 9 int n,k;10 11 int main()12 {13     scanf("%d",&t);14     while(t--)15     {16         vector<int>q[1100];17         scanf("%d%d",&n,&k);18         for(int i=1; i<=n; i++)19         {20             int m;21             scanf("%d",&m);22             for(int j=1; j<=m; j++)23             {24                 int x;25                 scanf("%d",&x);26                 q[i].push_back(x);27                 q[0].push_back(x);28             }29             sort(q[i].begin(),q[i].end());30         }31         sort(q[0].begin(),q[0].end());32         LL ans=0;33         for(int i=1; i<=n; i++)34         {35             for(int j=0; j<(int)q[i].size(); j++)36             {37                 int v=q[i][j];38                 LL num1=q[0].end()-lower_bound(q[0].begin(),q[0].end(),k-v+1);39                 LL num2=q[i].end()-lower_bound(q[i].begin(),q[i].end(),k-v+1);40                 ans+=num1-num2;41             }42         }43         printf("%I64d\n",ans/2);44     }45     return 0;46 }
View Code

 

hdu 5101 Select