链接:https://ac.nowcoder.com/acm/problem/13594
来源:牛客网
小L有严重的选择困难症。
早上起床后,需要花很长时间决定今天穿什么出门。
假设一共有k类物品需要搭配选择,每类物品的个数为Ai,每个物品有一个喜欢值Vj,代表小L对这件物品的喜欢程度。
小L想知道,有多少种方案,使得选出来的总喜欢值>M
需要注意,每类物品,至多选择1件,可以不选。
输入描述:
多组输入
每组数据第一行输入k M(k<=6,1<=M<=1e8),表示有多少类物品
接下来k行,每行以Ai(1<=Ai<=100)开头,表示这类物品有多少个,接下来Ai个数,第j个为Vj(1<=Vj<=1e8),表示小L对这类物品的第j个的喜欢值是多少。
输出描述:
每组输出一行,表示方案数
示例1
输入
复制
2 5
3 1 3 4
2 2 3
2 1
2 2 2
2 2 2
输出
复制
3
8
看给的数据大小我们是可以dfs的,在加一点贪心。
假如加上了i种物品的第j件物品,已经满足条件,那么剩下的数不论如何都是满足的,那么我们只需要求一下组合即可,—(每一种物品的数量+1)相乘之和就是答案。加一是因为可以不选。
其他的按照每一行每一列遍历即可,最后遍历一下不选的
#include<bits/stdc++.h>
using namespace std;
const int maxn=110;
typedef long long ll;
ll a[maxn][maxn];
ll cnt[maxn];
ll ans=0;
int n,k;
void dfs(int stedp,int sum)
{
if(sum>k)//剪枝
{
ll res=1;
for(int i=stedp;i<=n;i++)//对于余下的所有数的组合
{
res*=(cnt[i]+1);
}
ans+=res;
return;
}
if(stedp>n)
return ;
for(int i=1;i<=cnt[stedp];i++)//遍历每一行的没一列
{
dfs(stedp+1,sum+a[stedp][i]);
}
dfs(stedp+1,sum);//当前物品不选择
}
int main()
{
while(cin>>n>>k)
{
memset(a,0,sizeof a);
memset(cnt,0,sizeof cnt);
ans=0;
for(int i=1;i<=n;i++)
{
cin>>cnt[i];
for(int j=1;j<=cnt[i];j++)
cin>>a[i][j];
}
dfs(1,0);
cout<<ans<<endl;
}
return 0;
}