首页 > 代码库 > HDU 5101 Select --离散化+树状数组
HDU 5101 Select --离散化+树状数组
题意：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 __int64using namespace std;#define N 101107int 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;}
HDU 5101 Select --离散化+树状数组
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。