MYF

HDU 5101 Select

Posted on | In | | Visitors

题目链接

HDU 5101

题目类型：简单容斥 + 双指针

题目来源：BestCoder #17

题目分析

题目大意

给出$n$个班级以及小明的智商$k$，小明想要找两个队友，但是这两个队友的智商之和必须要大于小明的智商，并且要求这两个人不能是同一个班的，问有多少种组合？

解析

简单的容斥问题

想要找非同班的且智商大于$k$的组合，只需要找所有智商组合大于$k$的，再减去同班的智商大于$k$的组合即可。

对于每个班的组合，只需要在班内排序，然后使用双指针找到组合数累计。

对于全部的，只需要把所有智商排序，然后用双指针找到组合数。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <bitset>
#include <cstring>
#include <iostream>
#include <deque>
#include <algorithm>
#define Memset(a,val) memset(a,val,sizeof(a))
#define PI acos(-1)
#define PB push_back
#define MP make_pair
#define rt(n)       (i == n ? '\n' : ' ')
#define hi         printf("Hi----------\n")
#define IN freopen("input.txt","r",stdin);
#define OUT freopen("output.txt","w",stdout);
#define debug(x) cout<<"Debug : ---"<<x<<"---"<<endl;
#define debug2(x,y) cout<<"Debug : ---"<<x<<" , "<<y<<"---"<<endl;
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
const int mod=1000000007;
const int INF=0x3f3f3f3f;
const double eps=1e-8;
const int maxn=100000+50;
ll a[maxn];
int main(){
    int T,n,iq,m,tmp;
    scanf("%d",&T);
    while (T--) {
        scanf("%d%d",&n,&iq);
        
        ll tmp=0;   //同班的两人智商大于iq的情况
        int cnt=0;
        for (int i=0; i<n; i++) {
            scanf("%d",&m);
            int st = cnt,ed = cnt+m;
            for (int j=0; j<m; j++) {
                scanf("%lld",&a[cnt++]);
            }
            sort(a+st, a+ed);
            int l = st,r=ed-1;
            while (l<r) {
                if (a[r]+a[l]>iq) {
                    tmp+=r-l;
                    r--;
                }
                else
                    l++;
            }
        }
        
        ll total=0; //所有两者相加智商大于iq的情况
        int l=0,r=cnt-1;
        sort(a, a+cnt);
        while (l<r) {
            if (a[r]+a[l]>iq) {
                total+=r-l;
                r--;
            }
            else
                l++;
        }
        
        cout<<total-tmp<<endl;
    }
    return 0;
}