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

资源限制
时间限制：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;
}