标签:style blog http io color ar os sp for
题意:n 组,每组有一些值,求 在不同的两组中每组选一个使值的和大于k的方法数。
解法:n * Cnt[n] <= 1000*100 = 100000, 即最多10^5个人,所以枚举每个值x,求他的后面那些组中有多少数大于 k-x即可, 求有多少数大于k-x可以先求有多少数小于等于k-x,然后总数减一下即可。 可以用树状数组求。
先把所有数离散地存入一个树状数组中,然后每次枚举完一组的数后,将这组的数去掉。
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <cmath> #include <algorithm> #define lll __int64 using namespace std; #define N 101107 int c[N],num[1006][104],b[2*N],T[1007]; int lowbit(int x) { return x&-x; } void modify(int x,int val) { while(x <= N-10) { c[x] += val; x += lowbit(x); } } int getsum(int x) { int res = 0; while(x > 0) { res += c[x]; x -= lowbit(x); } return res; } int main() { int t,n,k,m,i,j; cin>>t; while(t--) { scanf("%d%d",&n,&k); int cnt = 0; for(i=1;i<=n;i++) { scanf("%d",&num[i][0]); for(j=1;j<=num[i][0];j++) scanf("%d",&num[i][j]), b[++cnt] = num[i][j]; sort(num[i]+1,num[i]+num[i][0]+1); } T[n+1] = 0; for(i=n;i>=1;i--) T[i] = T[i+1] + num[i][0]; sort(b+1,b+cnt+1); memset(c,0,sizeof(c)); for(i=1;i<=n;i++) { for(j=1;j<=num[i][0];j++) { int id = lower_bound(b+1,b+cnt+1,num[i][j])-b; modify(id,1); } } lll sum = 0; for(i=1;i<=n;i++) { for(j=1;j<=num[i][0];j++) { int id = lower_bound(b+1,b+cnt+1,num[i][j])-b; modify(id,-1); } for(j=1;j<=num[i][0];j++) { int now = num[i][j],id; if(k-now < 0) id = 0; else { id = lower_bound(b+1,b+cnt+1,k-now)-b; if(b[id] != k-now) id--; } sum += T[i+1]-getsum(id); } } cout<<sum<<endl; } return 0; }
标签:style blog http io color ar os sp for
原文地址:http://www.cnblogs.com/whatbeg/p/4084112.html