题目大意:第一行n,k,分别代表n个组和当前人的IQ是多少,问从不同的组里选出两个数的和>k的方法数是多少少。
思路:按照题目的思路 不容易想到好点的算法解觉,我们可以转化一下
不同组任意两数和=不分组任意两数和-同一组任意两数和。 那么同一组的任意两数和大于k,可以通过排序得到,并且通过lower_bound()查找得到。
#include<algorithm>
#include<cmath>
#include<queue>
#include<map>
#define ll long long
#define inf 0x3f3f3f3f
#define ls l,mid,rt<<1
#define rs mid+!,r,rt<<1|1
#include <vector>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
ll n,m;
int main(){
ll n,m,i,j,t,k;
ll cla;
vector<ll>q[1010];
scanf("%lld",&cla);
while(cla--){
for(i = 0;i <1010;++ i ){
q[i].clear();
}
ll x;
scanf("%lld%lld",&n,&k);
for(i = 1;i <= n;++ i){
scanf("%lld",&m);
for(j = 0;j < m; ++ j){
scanf("%I64d",&t);
q[0].push_back(t);
q[i].push_back(t);
}
sort(q[i].begin(),q[i].end());
}
sort(q[0].begin(),q[0].end());
ll ans = 0;
for( i = 1;i<= n;++ i){
for(j = 0;j < q[i].size();++ j ){
ll s1,s2;
ll v = q[i][j];
s2 =q[i].end() - lower_bound(q[i].begin(),q[i].end(),k-v+1);
s1 =q[0].end() - lower_bound(q[0].begin(),q[0].end(),k-v+1 );
ans += s1 - s2;
}
}
printf("%lld\n",ans / 2);
}
return 0;
}