刷题记录:牛客NC50243小木棍

传送门:牛客

题目描述:

乔治有一些同样长的小木棍，他把这些木棍随意砍成几段，直到每段的长都不超过50。现在，他想把小木棍
拼接成原来的样子，但是却忘记了自己开始时有多少根木棍和它们的长度。给出每段小木棍的长度，编程帮
他找出原始木棍的最小可能长度。
输入:
9
5 2 1 5 2 1 5 2 1
输出:
6

emmm,这是一道搜索题,刚开始我还以为是二分答案…结果后来感觉推不出单调性,

主要思路:

1. 延续我们二分的思路(虽然不能二分),我们会发现我们的最小可能长度只可能是max{小木棍}~sum{小木棍},然而我们再看一下这道题的数据范围似乎也不大,就更加给了我们搜索的信心.
2. 对于每一根木棍,我们先进行一波排序(从大到小),然后开始枚举我们的开始的木棍大小,此时不难发现我们枚举的最开始的木根的大小应该是一个整数,并且应该是被我们的总长度所整除的(因为最开始的所有木棍的大小是相同的),此时我们就很自然的进行了一次剪枝优化.
3. 然而假设我们就不再进行任何优化就开始爆搜的话,我们会很自然的得到一个TLE.(不要问为什么我知道,问就是被卡了)
   4.我们再仔细的思考,会发现其实还是可以进行一下的剪枝的(emmm,这部分剪枝不太好讲,在代码中结合注释讲解吧)

下面是具体的代码部分:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
#include <deque>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
int n;int a[maxn];int vis[maxn];
int cnt=0;int l=-inf,r=0;
int dfs(int now_l,int tot_l,int pos) {
//now_l代表当前木棍的剩余长度,tot_l代表每根木棍最终的长度,pos代表枚举的木棍标号,因为是从大
//到小进行排序过得,所以如果当前不行,那么之前的旧更加不行了,后面的可能可以
	if(now_l==0) {
		cnt++;//cnt代表木棍的个数,当cnt*tot_l=木棍的总和时,代表我的当前编号是可行的
		now_l=tot_l;//换下一根木棍
		pos=1;//位置当然要变成一
	}
	if(cnt*tot_l==r) return true;
	for(int i=pos;i<=n;i++) {//对我们剩余的小木棍进行一个枚举
		int temp=cnt;//注意,这里需要记录我们的cnt的初始值,刚开始我没有记录,结果导致一直wa两个点
		//因为假设我们不将cnt进行回溯的话,我们的cnt会因为之后的"不行"的搜索而改变值,然而我们肯定是不希望它进行改变的
		if(vis[i]) continue;
		if(a[i]>now_l) continue;//如果当前的木棍的长度大于我们所需的长度
		vis[i]=1;
		if(dfs(now_l-a[i],tot_l,i+1)) return true;
		vis[i]=0;
		cnt=temp;
		if(now_l==a[i]||now_l==tot_l) break;
		//注意这里的剪枝,一共有两部分的剪枝:
		//第一部分的剪枝解释:这一步的剪枝是说,当我们当前的木棍和我们此时大木棍的剩余长度相同时
		//但是我们此时经过这一次的搜索发现竟然还是不行,那么我们此时的目标长度肯定是不行的,为什么这样说呢
		//因为这根木棍最后肯定是要接在一个大木棍的后面的,而这根大木棍的长度也肯定是和我们此时枚举的长度相同的,也就是说这个小木棍肯定是要接在我们的这根木棍后面的,但是还是不行,说明此时的长度肯定是错误的,所以直接结束我们的枚举即可
		//对于我们的第二个剪枝,也是就当前的长度和我们刚开始的长度一样,也就是说此时我们的木棍竟然刚开始就不能接上其他的木棍,那么小木棍也就无法接上其他的大木棍后面了,也就是当前的木棍太长了,导致大于我的所定的长度,所以是不行的
		while(a[i]==a[i+1]) i++;
		这里是第三部分的剪枝,假设我们到达了这里,说明我们当前小木棍是不能加入的,但是如果后面的小木棍和现在的小木棍相等的话呢,就说明下一根木棍显然是不能加入的,所以直接排除即可
	}
	return false;
}

bool cmp(int a,int b) {
	return a>b;
}
int main() {
	memset(vis,0,sizeof(vis));
	n=read();
	for(int i=1;i<=n;i++) {
		a[i]=read();
		l=max(l,a[i]);
		r+=a[i];
	}
	sort(a+1,a+n+1,cmp);
	for(int i=l;i<=r;i++) {
		cnt=0;
		if(r%i!=0) continue;//最显而易见的一个剪枝
		if(dfs(i,i,1)) {
			cout<<i<<endl;
			return 0;
		}
	}
	return 0;
}