Java实现 蓝桥杯 算法提高 天天向上 动态规划

试题 算法提高 天天向上

问题描述

A同学的学习成绩十分不稳定，于是老师对他说：“只要你连续4天成绩有进步，那我就奖励给你一朵小红花。”可是这对于A同学太困难了。于是，老师对他放宽了要求：“只要你有4天成绩是递增的，我就奖励你一朵小红花。”即只要对于第i、j、k、l四天，满足i<j<k<l并且对于成绩wi<wj<wk<wl，那么就可以得到一朵小红花的奖励。现让你求出，A同学可以得到多少朵小红花。

输入格式
　　第一行一个整数n，表示总共有n天。第二行n个数，表示每天的成绩wi。
输出格式
　　一个数，表示总共可以得到多少朵小红花。
样例输入
6
1 3 2 3 4 5
样例输出
6
数据规模和约定
　　对于40%的数据，n<=50；
　　对于100%的数据，n<=2000，0<=wi<=109。

import java.math.BigInteger;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
       int[] a = new int[n+1];
       for (int i = 1; i < a.length; i++) {
		a[i] = sc.nextInt();
	}
       BigInteger i = dp(a,n);
       
       System.out.println(i);
	}

	private static BigInteger dp(int[] a, int n) {
		// TODO Auto-generated method stub
		BigInteger dp[][] = new BigInteger[4][n+1];
		for (int i = 0; i < dp.length; i++) {
			for (int j = 0; j <= n; j++) {
				dp[i][j] = new BigInteger("0");
			}
		}
		
		for (int i = 1; i <= n ; i++) {
			dp[0][i] = new BigInteger("1");
		}
		
		for (int i = 1; i < dp.length; i++) {
			for (int j = i+1; j <= n; j++) {
				
					for(int k = j-1; k > 0; k--){
						if(a[j] > a[k]){
							dp[i][j] = dp[i][j].add(dp[i-1][k]) ;
						}
					}
				
			}
		}
		
		BigInteger res = new BigInteger("0");
		for (int i = 0; i < dp[0].length; i++) {
			res = res.add(dp[3][i]);
		}
		return res;
	}

}