选择困难症

发布时间: 2017年7月4日 12:44   最后更新: 2017年7月4日 12:45   时间限制: 5000ms   内存限制: 128M

3 3
2 1 2
2 1 2
2 1 2

10

代码:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
#define int long long
const int MAX = 1e7+8;
int k,M;
int a[7][107];
int cnt[6];
int arr[MAX];
int brr[MAX];
int cnta = 0;
int cntb = 0;
void dfs(int array[],int& cntp,int step,int end,int sum){if(step == end){array[cntp++] = sum;return ;}for(int i = 0;i <= cnt[step];i++){dfs(array,cntp,step+1,end,sum + a[step][i]);}}
main(){while(scanf("%lld%lld",&k,&M) != EOF){memset(cnt,0,sizeof(cnt));cnta = cntb = 0;for(int i = 0;i < k;i++){scanf("%lld",&cnt[i]);for(int j = 0;j < cnt[i];j++){scanf("%lld",&a[i][j]);}a[i][cnt[i]] = 0;}int mid = k / 2;int last = 0;dfs(arr,cnta,0,mid,0);dfs(brr,cntb,mid,k,0);sort(brr,brr + cntb);int ans = 0;for(int i = 0;i < cnta;i++){ans += brr + cntb - upper_bound(brr,brr + cntb,M - arr[i]);}printf("%lld\n",ans);}return 0;
}