木棍【dfs搜索优化】

木棒

题目描述

在这里插入图片描述

输入样例:

9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0

输出样例:

6
5

【思路】

优化
在这里插入图片描述

在这里插入图片描述

【AC代码】

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 70;

int w[N], sum, length, n;
bool st[N];

bool dfs(int u, int cur, int start) {	//当前拼接木棒个数,当前木棒长度 ,从第几个已有木棒进行枚举搜索 
	if(u * length == sum)	return true;// 如果当前木棒的数量 * 长度 = sum,则找到了答案,直接返回 true
	if(cur == length) return dfs(u + 1, 0, 0);// 如果当前木棒的长度等于了 length,则新开一根木棒
	// 否则枚举从 start 开始枚举将哪根木棍拼到当前木棒中
	for(int i = start; i < n; i ++) {
		//若已经搜索过 
		if(st[i] || cur + w[i] > length)	continue;
		//若没有搜索过
		st[i] = true;
		// 继续搜索当前木棒的下一根木棍拼哪根
		if(dfs(u, cur + w[i], i + 1))	return true;
		//恢复现场
		st[i] = false;
		
		// 执行到这 dfs(u, s + w[i], i + 1) 为 false: 说明当前木棍搜索失败了
		 
		//继续进行剪枝优化
		
		//如果第一根或者最后一根搜索失败,那么一定失败
		if(cur == 0 || cur + w[i] == length)	return false;
		int j = i;
// 		如果当前木棍没有搜到方案,则跳过所有长度相等的木棍
		while(j < n && w[j] == w[i])	j ++;
		i = j - 1; 
	}	
	//若是上边没有匹配成功,则返回失败
	return false;
}

int main() {
	while(cin >> n, n) {
		memset(st, 0, sizeof st);
		sum = 0, length = 1; // 每根木棒的长度从 1 开始枚举,从小到大枚举,题目要求分组最多,那么第一个满足条件的就是答案
		for(int i = 0; i < n; i ++) {
			cin >> w[i];
			sum += w[i];
		}
		//从大到小枚举,减少搜索分支 
		sort(w, w + n, greater<int>());
		
		while(true) {// 循环一定会退出,最坏情况下所有木棍拼一根木棒
			if(sum % length == 0 && dfs(0, 0, 0)) {
				cout << length << endl;
				break;
			}
			length++;// 当前 length 无法得到答案,则长度 + 1
		}
	}
	return 0;
} 




  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
A:木棍还原问题是一种常见的深度优先算法问题。具体的思路是,先将破碎的木棍抽象成一个图形,然后从任意一根木棍开始,以逆时针的顺序将其端点遍历,并将遍历过的木棍断开,变成两根更短的木棍。然后从任意一条与当前木棍相邻的木棍再次开始遍历,直到所有的木棍都被拼接完成或者无法拼接。 具体步骤如下: 1. 选择一根木棍,从其中任意一端开始遍历它的邻接木棍。 2. 对于每个邻接木棍,重复1,直到所有邻接木棍被遍历过。 3. 如果当前木棍的两端都没有邻接木棍,则将其标记为已经访问,从当前栈中弹出并返回上一级调用。 4. 否则,选择一个未曾遍历的邻接木棍,将其作为当前木棍,将其标记为已访问并进入栈中,从1开始。 在实现深度优先算法时,需要注意以下几点: 1. 应该使用栈来维护当前遍历路径。 2. 应该记录每个木棍是否已经访问过,避免重复访问。 3. 应该检测当前木棍的邻接木棍是否能够拼接,如果不能,应该回溯到上一个节点尝试其他路径。 代码示例: ``` def dfs(stick, path): if len(path) == len(stick): # 所有木棍已经还原完成 return True for i in range(len(stick)): if not visited[i]: # 当前木棍未被访问 visited[i] = True path.append(stick[i]) for j in range(i+1, len(stick)): if not visited[j] and sum(path) + stick[i] + stick[j] <= total_length: visited[j] = True path.append(stick[j]) if sum(path) == total_length: # 可以完成拼接 return True if dfs(stick, path): # 搜索下一级节点 return True path.pop() visited[j] = False path.pop() visited[i] = False return False ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值