问题 1551: [蓝桥杯][算法提高VIP]种树（dfs）

题目链接：点击这里

$n \leq 30$，暴搜即可。

#include<iostream>
#include<algorithm>
#include<cstdio>

using namespace std;
typedef long long ll;
const int N = 50;

int n, m, ans = -1e9;
int a[N];
bool st[N];

void dfs(int idx, int now, int sum)
{
	if(now == m)
	{
		ans = max(ans, sum);
		return;
	}
	
	if(idx > n)	return;
	
	if(idx==n)			//第n号要单独判断 
	{
		if(now==m-1 && !st[1] && !st[n-1])
		{
			sum += a[n];
			ans = max(ans, sum);
		}
		return;
	}
	
	st[idx] = true;
	dfs(idx + 2, now + 1, sum + a[idx]);	//选idx号，那么下一个只能从idx+2号开始选	

	st[idx] = false;			//恢复现场 
	
	dfs(idx + 1, now, sum);		//不选idx号，那么下一个可以从idx+1号开始选 
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; ++i)	scanf("%d", &a[i]);
	
	if(n < m * 2)
	{
		puts("Error!");
		return 0;
	}
	
	dfs(1, 0, 0);
	
	printf("%d\n", ans);
	
	return 0;
}