长沙理工大学第十二届ACM大赛L 选择困难症 (剪枝暴搜)

链接:https://ac.nowcoder.com/acm/contest/1/L
来源:牛客网

选择困难症
时间限制:C/C++ 3秒,其他语言6秒
空间限制:C/C++ 131072K,其他语言262144K
64bit IO Format: %lld
题目描述
小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,当dfs的x参数大于m的时候,直接计算答案的贡献,不继续搜,这样的优化就可以AC本题,但是题目数据太弱了,如果是 6*100个1的话,而m是1e8 ,那么这种写法是直接卡到TLE的,所以是一个有问题的题目,大家随便做做,开心就好。

细节见代码:

#include <iostream>
	#include <cstdio>
	#include <cstring>
	#include <algorithm>
	#include <cmath>
	#include <queue>
	#include <stack>
	#include <map>
	#include <set>
	#include <vector>
	#include <iomanip>
	#define ALL(x) (x).begin(), (x).end()
	#define rt return
	#define sz(a) int(a.size())
	#define all(a) a.begin(), a.end()
	#define rep(i,x,n) for(int i=x;i<n;i++)
	#define repd(i,x,n) for(int i=x;i<=n;i++)
	#define pii pair<int,int>
	#define pll pair<long long ,long long>
	#define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
	#define MS0(X) memset((X), 0, sizeof((X)))
	#define MSC0(X) memset((X), '', sizeof((X)))
	#define pb push_back
	#define mp make_pair
	#define fi first
	#define se second
	#define eps 1e-6
	#define gg(x) getInt(&x)
	#define db(x) cout<<"== [ "<<x<<" ] =="<<endl;
	using namespace std;
	typedef long long ll;
	ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
	ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
	ll powmod(ll a,ll b,ll MOD){ll ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
	inline void getInt(int* p);
	const int maxn=1000010;
	const int inf=0x3f3f3f3f;
	/*** TEMPLATE CODE * * STARTS HERE ***/


	ll m;
	int k;
	ll a[10][200];
	ll num[10];
	ll mulnum[10];
	ll ans;
	void dfs(int pos,ll x )
	{
		if(pos>k)
		{
			return ;
		}
		for(int i=1;i<=num[pos];i++)
		{
			if(x+a[pos][i]>m)
			{
				ans+=(num[pos]-i+1)*mulnum[pos+1];
				return ;
			}
			dfs(pos+1,x+a[pos][i]);
		}
	}
	int main()
	{
	    //freopen("D:\code\text\input.txt","r",stdin);
		//freopen("D:\code\text\output.txt","w",stdout);
		
		while(cin>>k>>m)
		{
			repd(i,1,k)
			{
				cin>>num[i];
				repd(j,1,num[i])
				{
					cin>>a[i][j];
				}
				a[i][num[i]++]=0;
				sort(a[i]+1,a[i]+1+num[i]);
			}
			mulnum[k+1]=1ll;
			for(int i=k;i>=1;i--)
			{
				mulnum[i]=mulnum[i+1]*num[i];
			}
			ans=0ll;
			dfs(1,0ll);
			cout<<ans<<endl;
		}
		
		
		
	    return 0;
	}

	inline void getInt(int* p) {
	    char ch;
	    do {
	        ch = getchar();
	    } while (ch == ' ' || ch == '
');
	    if (ch == '-') {
	        *p = -(getchar() - '0');
	        while ((ch = getchar()) >= '0' && ch <= '9') {
	            *p = *p * 10 - ch + '0';
	        }
	    }
	    else {
	        *p = ch - '0';
	        while ((ch = getchar()) >= '0' && ch <= '9') {
	            *p = *p * 10 + ch - '0';
	        }
	    }
	}


本博客为本人原创,如需转载,请必须声明博客的源地址。 本人博客地址为:www.cnblogs.com/qieqiemin/ 希望所写的文章对您有帮助。
原文地址:https://www.cnblogs.com/qieqiemin/p/11000600.html