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就是答案。

 

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
#define LL __int64
using namespace std;

int t;
int n,k;

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        vector<int>q[1100];
        scanf("%d%d",&n,&k);
        for(int i=1; i<=n; i++)
        {
            int m;
            scanf("%d",&m);
            for(int j=1; j<=m; j++)
            {
                int x;
                scanf("%d",&x);
                q[i].push_back(x);
                q[0].push_back(x);
            }
            sort(q[i].begin(),q[i].end());
        }
        sort(q[0].begin(),q[0].end());
        LL ans=0;
        for(int i=1; i<=n; i++)
        {
            for(int j=0; j<(int)q[i].size(); j++)
            {
                int v=q[i][j];
                LL num1=q[0].end()-lower_bound(q[0].begin(),q[0].end(),k-v+1);
                LL num2=q[i].end()-lower_bound(q[i].begin(),q[i].end(),k-v+1);
                ans+=num1-num2;
            }
        }
        printf("%I64d\n",ans/2);
    }
    return 0;
}