[蓝桥杯][算法提高VIP]种树（dfs）

A城市有一个巨大的圆形广场，为了绿化环境和净化空气，市政府决定沿圆形广场外圈种一圈树。园林部门得到指令后，初步规划出n个种树的位置，顺时针编号1到n。并且每个位置都有一个美观度Ai，如果在这里种树就可以得到这Ai的美观度。但由于A城市土壤肥力欠佳，两棵树决不能种在相邻的位置（i号位置和i+1号位置叫相邻位置。值得注意的是1号和n号也算相邻位置！）。
最终市政府给园林部门提供了m棵树苗并要求全部种上，请你帮忙设计种树方案使得美观度总和最大。如果无法将m棵树苗全部种上，给出无解信息。

数据规模和约定

对于全部数据，满足1< =m< =n< =30；
其中90%的数据满足m< =n< =20
-1000< =Ai< =1000

输入
输入的第一行包含两个正整数n、m。
第二行n个整数Ai。
输出
输出一个整数，表示最佳植树方案可以得到的美观度。如果无解输出“Error!”，不包含引号。
样例输入
7 3
1 2 3 4 5 6 7
样例输出
15
思路：深搜+回溯。数据量不大，可以直接搜。跟之前的一道填空题类似。但是Error注意还有一个！，白wa一发。
代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int maxx=31;
int val[maxx];
int vis[maxx];
int n,m;

inline int check(int &sum)
{
	if(vis[1]&&vis[n]) return 0;
	int cnt=0;
	for(int i=1;i<n;i++) if(vis[i]&&vis[i+1]) return 0;
	for(int i=1;i<=n;i++) if(vis[i]) cnt++,sum+=val[i];
	return cnt==m;
}
inline void dfs(int x,int &ans)
{
	if(x==n+1)
	{
		int sum=0;
		if(check(sum)) ans=max(ans,sum);
		return ;
	}
	if(vis[x-1]==0)
	{
		vis[x]=1;
		dfs(x+1,ans);
		vis[x]=0;
	}
	dfs(x+1,ans);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&val[i]);
	int ans=0;
	memset(vis,0,sizeof(vis));
	dfs(1,ans);
	if(ans==0) cout<<"Error!"<<endl;
	else cout<<ans<<endl;
	return 0;
}

努力加油a啊，(^o)/~

来源：CSDN

作者：starlet_kiss

链接：https://blog.csdn.net/starlet_kiss/article/details/104789222

标签

蓝桥杯