Java实现 蓝桥杯VIP 算法提高 种树

算法提高 种树

时间限制：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

import java.util.Scanner;

public class 种树 {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		int n,m,a[],i,j,dp[][][][],num;
		n=Integer.parseInt(sc.next());
		m=Integer.parseInt(sc.next());
		num=m+1;
		if(m>n/2)System.out.println("Error!");
		else
		{
			a=new int[n];
			for(i=0;i<n;i++)a[i]=Integer.parseInt(sc.next());
			dp=new int[n][2][2][num];
			dp[0][1][0][1]=a[0];
			dp[0][0][1][1]=a[0];
			for(i=1;i<n;i++)
			{
				for(j=1;j<num;j++)
				{
					dp[i][0][0][j]=Integer.MIN_VALUE/2;
					dp[i][1][0][j]=Integer.MIN_VALUE/2;
					dp[i][0][1][j]=Integer.MIN_VALUE/2;
					dp[i][1][1][j]=Integer.MIN_VALUE/2;
				}
			}
			if(n>1)
			{
				dp[1][0][1][1]=a[1];
				dp[1][1][0][1]=a[0];
			}
			if(n>2)
			{
				dp[2][1][0][1]=a[0];
				dp[2][0][1][1]=a[2];
				if(m>2)dp[2][1][1][2]=a[0]+a[2];
				dp[2][0][0][1]=a[1];
			}
			for(i=3;i<n;i++)
			{
				for(j=0;j<num;j++)
				{
					dp[i][0][0][j]=max(dp[i-1][0][0][j],dp[i-1][0][1][j]);
					dp[i][1][0][j]=max(dp[i-1][1][0][j],dp[i-1][1][1][j]);
					if(j>1&&j*2<=i+1)dp[i][0][1][j]=dp[i-1][0][0][j-1]+a[i];
					else if(j==1)dp[i][0][1][j]=a[i];
					if(j>1)dp[i][1][1][j]=dp[i-1][1][0][j-1]+a[i];
				}
			}
			System.out.println(max(dp[n-1][0][0][m], max(dp[n-1][0][1][m], dp[n-1][1][0][m])));
		}
	}

	private static int max(int i, int j) {
		// TODO Auto-generated method stub
		if(i>j)return i;
		else return j;
	}

}