每日一题第七十八期洛谷小木棍-CSDN博客

本文链接：https://blog.csdn.net/2301_80882026/article/details/137482688

本文介绍了一个编程问题，乔治有一堆被砍切的小木棍，需要通过给定每段木棍的长度，找到重新组合成原长度的最小原始木棍总长度。文章提供了使用深度优先搜索（dfs）的解决方案。

摘要由CSDN通过智能技术生成

小木棍

题目背景

本题不保证存在可以通过满足本题数据范围的任意数据做法。可以通过此题的程序不一定完全正确（算法时间复杂度错误、或不保证正确性）

本题为搜索题，本题不接受 hack 数据。关于此类题目的详细内容

题目描述

乔治有一些同样长的小木棍，他把这些木棍随意砍成几段，直到每段的长都不超过 $50$ 。

现在，他想把小木棍拼接成原来的样子，但是却忘记了自己开始时有多少根木棍和它们的长度。

给出每段小木棍的长度，编程帮他找出原始木棍的最小可能长度。

输入格式

第一行是一个整数 $n$ ，表示小木棍的个数。
第二行有 $n$ 个整数，表示各个木棍的长度 $a_i$ 。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

9
5 2 1 5 2 1 5 2 1

样例输出 #1

提示

对于全部测试点， $\leq n \leq 65$ ， $\leq a_i \leq 50$ 。

AC代码：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<string>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<numeric>
#include<iomanip>
#define endl '\n'
//#define x first
//#define y second
using namespace std;
typedef long long ll;
typedef pair<int, int>PII;
const int N=3e5+10;
const int MOD=9901;
const int INF=0X3F3F3F3F;
const int dx[]={-1,1,0,0,-1,-1,+1,+1};
const int dy[]={0,0,-1,1,-1,+1,-1,+1};
const int M = 1e6 + 10;

int n;
int w[N];
bool st[N];
int len, sum;
bool dfs(int u, int cur, int start)
{
	if(u * len == sum) return true;
	if(cur == len) return dfs(u + 1, 0, 0);
	for(int i = start; i < n; i ++)
	{
		if(st[i]) continue;
		if(cur + w[i] <= len)
		{
			st[i] = true;
			if(dfs(u, cur + w[i], i + 1)) return true;
			st[i] = false;
		}
		if(!cur || cur + w[i] == len) return false;
		int j = i + 1;
		while(j < n && w[i] == w[j]) j ++;
		i = j - 1;
	}
	return false;
}
int main()
{
	while(cin >> n && n){
		sum = 0, len = 0;
		for(int i = 0; i < n; i ++)
		{
			cin >> w[i];
			sum += w[i];
			len = max(len, w[i]);
		}
		sort(w, w + n, greater<int>());
		memset(st, 0 , sizeof st);
		while(true){
			if(sum % len == 0 && dfs(0, 0, 0))
			{
				cout << len << endl;
				break;
			}
			len ++;
		}
	}
	return 0;
}