zcmu—2106: 种树（递归dfs）

题目描述

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

输入

输入的第一行包含两个正整数n、m。
第二行n个整数Ai。

输出

输出一个整数，表示最佳植树方案可以得到的美观度。如果无解输出“Error!”，不包含引号。

样例输入

7 3

1 2 3 4 5 6 7

样例输出

15

代码：

#include<bits/stdc++.h>
using namespace std;
int a[50];
int way[50];
int n,m,mx,w=0;
void dfs(int x ,int n1,int sum){
	if(n1==n){
		mx=max(mx,sum);
		return ;
	}
	if(x>m)return ;
	if(way[x-1]==0&&way[x+1]==0){
		if(x==1)way[m+1]=1;
		way[x]=1;
		dfs(x+1,n1+1,sum+a[x]);
		way[x]=0;
		if(x==1)way[m+1]=0;
	}
	dfs(x+1,n1,sum);
}
int main(){
	while(~scanf("%d %d",&m,&n)){
		memset(way,0,sizeof(way));
        for(int i=1;i<=m;i++){
        	scanf("%d",&a[i]);
		}
		if(m/2<n)printf("Error!\n");
		else{
		mx=0;
		dfs(1,0,0);
	    printf("%d\n",mx);
	}
	}
}