蓝桥杯 ADV-139 种树

  算法提高 种树  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　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

样例输入

7 4
1 2 3 4 5 6 7

样例输出

Error!

数据规模和约定

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

分析：难点在于处理环状结构。用visit数组来存储访问情况，通过(index - 1 + n) % n与(index + 1) % n来查看index在环中相邻的两树是否被访问过。之后用递归处理即可。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int n, m;
vector<int> v;
vector<int> visit;
int ans = -1;
void f(int num, int cnt, int index) {
	if (cnt == m) {
		ans = max(ans, num);
		return;
	}
	if (index >= n) {
		return;
	}
	if (!visit[(index - 1 + n) % n] && !visit[(index + 1) % n]) {
		if (!visit[index]) {
			visit[index] = 1;
			f(num + v[index], cnt + 1, index + 1);
			visit[index] = 0;
		}
	}
	f(num, cnt, index + 1);
}
int main() {
	cin >> n >> m;
	if (m > n / 2) {
		cout << "Error!" << endl;
	} else {
		v.resize(n);
		visit = v;
		for (int i = 0; i < n; i++) {
			cin >> v[i];
		}
		f(0, 0, 0);
		cout << ans << endl;
	}
	return 0;
}