hdu 5101 Select

某场BestCoder的题,题意是说从n个小组里的某两个小组中各选一个人,要求所选的两个人的IQ和必须大于k。问符合的情况有多少种。

答案=从所有数中选择的两个加和大于k的数的方案数-在同一个集合中选择的两个加和大于k的数的方案数
而对于同一个集合中选择的两个加和大于k的方案数是可以直接排序然后利用单调性快速统计出来的。
		————出题者

PS:注意答案可以超int,所以必须longlong...马丹当时因为这个跪了

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
#include <cmath>
#include <map>
#include <vector>
using namespace std;
typedef long long LL;
vector<LL> b;
int t, n;
LL a[1010][110];
int m[1010];
int main() {
    LL flag;
    while(~scanf("%d", &t)) {
        while(t--) {
            scanf("%d%I64d", &n, &flag);
            b.clear();
            int sum_num = 0;
            for(int i = 0; i < n; i++) {
                scanf("%d", &m[i]);
                sum_num += m[i];
                for(int j = 0; j < m[i]; j++) {
                    scanf("%I64d", &a[i][j]);
                    b.push_back(a[i][j]);
                }
                sort(a[i], a[i] + m[i]);
            }
            sort(b.begin(), b.end());
            LL ans = 0, x, y;
            for(int i = 0; i < n; i++) {
                for(int j = 0; j < m[i]; j++) {
                    LL temp = a[i][j];
                    int tmp1 = lower_bound(b.begin(), b.end(), flag-temp+1) - b.begin();
                    x = (int)b.size() - tmp1;
                    if(temp * 2 > flag) x--;
                    int tmp2 = lower_bound(a[i], a[i]+m[i], flag-temp+1) - a[i];
                    y = m[i] - tmp2;
                    if(temp * 2 > flag) y--;
                    ans += (x - y);
                }
            }
            printf("%I64d\n", ans / 2);
        }
    }
    return 0;
}


posted @ 2014-11-08 21:51  gaoxiang36999  阅读(188)  评论(0编辑  收藏  举报