蓝桥杯试题算法训练粘木棍

赤坂

已于 2022-02-27 21:04:02 修改

阅读量790

点赞数 2

文章标签：算法深度优先蓝桥杯

于 2022-02-21 13:44:27 首次发布

本文链接：https://blog.csdn.net/qq_33905610/article/details/123045237

版权

该博客介绍了一种使用深度优先搜索（DFS）解决如何将N根木棍粘贴成M个长木棍，使最长和最短长度差最小的问题。代码示例中，通过递归遍历所有可能的木棍组合，并在每一步中计算当前最大值和最小值的差值，以找到最小差距。最终输出这个最小差距。

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

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　有N根木棍，需要将其粘贴成M个长木棍，使得最长的和最短的的差距最小。
输入格式
　　第一行两个整数N,M。
　　一行N个整数，表示木棍的长度。
输出格式
　　一行一个整数，表示最小的差距
样例输入
3 2
10 20 40
样例输出
10
数据规模和约定
　　N, M<=7
思路：
数据量较小，可以直接暴力枚举所有状态判断

使用DFS+穷举的方法，递减数组长度，每次取两个木棍合并，计算每一个状态的最大值减最小值，通过全局变量记录所有解并取最小值。

#include <bits/stdc++.h>
using namespace std;
int a[10];//存放木棍
int n, m, d;//d用来存储解得最小值

void  dfs(int k)
{
	if (k == m)//剩余木棍数理已经达到了目标
	{
		int max1 = 0;
		int min1 = 9999;
		for (int i = 0; i < k; i++)//遍历找最值
		{
			if (max1 < a[i])
				max1 = a[i];
			if (min1 > a[i])
				min1 = a[i];
		}
		d = min(d,(max1 - min1));//取最小的解给d
		return;
	}
	for (int i = 0; i < k; i++)//循环遍历的第一根木棍
	{
		for (int j = i + 1; j < k; j++)//第二根木棍
		{
			a[i] += a[j];//两根木棍合成新木棍
			swap(a[j], a[k - 1]);//使用过的木棍放在队尾
			dfs(k - 1);//向下一层遍历
			swap(a[j], a[k - 1]);//恢复
			a[i] -= a[j];
		}
	}
}
int main()
{
	cin >> n >> m;
	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
	}
	d = 9999;
	dfs(n);
	cout << d;
	return 0;
}