hdu 5101 n集合選2個不同集合數使和大於k

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

給n個集合，選擇兩個來自不同集合的數，加和大於k，問有多少種選擇方案。

答案=從所有數中選擇的兩個加和大於k的數的方案數-在同一個集合中選擇的兩個加和大於k的數的方案數
而對於同一個集合中選擇的兩個加和大於k的方案數是可以直接排序然後利用單調性快速統計出來的。

注意upper_bound的應用和ans要使用long long因爲10^5*10^5/2超界限了..

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <map>
#include <cstring>
#include <algorithm>
#define RD(x) scanf("%d",&x)
#define RD2(x,y) scanf("%d%d",&x,&y)
#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define clr0(x) memset(x,0,sizeof(x))
#define clr1(x) memset(x,-1,sizeof(x))
using namespace std;
typedef long long LL;
const int maxn = 1005,maxm = 105;
int p[maxn][maxm],n,k,s[maxn*maxm];

int main()
{
    //cout<<(int)2147483647<<endl;
    int _;RD(_);
    while(_--)
    {
        int cnt = 0;
        LL ans = 0;
        RD2(n,k);
        for(int i = 1;i <= n;++i){
            RD(p[i][0]);
            for(int j = 1;j <= p[i][0];++j){
                RD(p[i][j]);
                s[cnt++] = p[i][j];
            }
        }
        sort(s,s+cnt);
        for(int i = 0;i < cnt;++i){
            ans += (s + cnt - upper_bound(s,s+cnt,k - s[i]));
        }
        for(int i = n;i >= 1;--i){
            sort(p[i]+1,p[i] + p[i][0] + 1);
            for(int j = 1;j <= p[i][0];++j){
                ans -= (p[i] + p[i][0] + 1 - upper_bound(p[i]+1,p[i]+p[i][0]+1,k - p[i][j]));
            }
        }
        printf("%I64d\n",ans/2);
    }
    return 0;
}