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;
	}


}

原文地址:https://www.cnblogs.com/a1439775520/p/13078340.html