Java 算法 摆动序列

题目描述

如果一个序列满足下面的性质，我们就将它称为摆动序列：
　　1. 序列中的所有数都是不大于k的正整数；
　　2. 序列中至少有两个数。
　　3. 序列中的数两两不相等；
　　4. 如果第i – 1个数比第i – 2个数大，则第i个数比第i – 2个数小；如果第i – 1个数比第i – 2个数小，则第i个数比第i – 2个数大。
　　比如，当k = 3时，有下面几个这样的序列：
　　1 2
　　1 3
　　2 1
　　2 1 3
　　2 3
　　2 3 1
　　3 1
　　3 2
　　一共有8种，给定k，请求出满足上面要求的序列的个数。

输入
输入描述:
　　输入包含了一个整数k。（k<=20）
输入样例:
3

输出

输出描述:
　　输出一个整数，表示满足要求的序列个数。
输出样例:
8

HINT:时间限制：1.0s 内存限制：512.0MB

解题思路

就是一个动态规划的问题，j=2即两个数组成时,随便排列组合都是摆动序列,即有C(i , 2) = A(i,2) = i * (i - 1) 则C(i,j) = C(i - 1，j - 1) + C(i - 1, j)。

代码

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        int n=scanner.nextInt();
        int[][] a=new int[n+1][n+1];
        for(int i=2;i<a.length;i++){//两个数无论怎么组合都是摆动序列
            a[i][2]=i*(i-1);
            a[i][i]=2;
        }
        //a[i][j]表示K = i时由j个数组成的摆动序列有多少种
        for(int i=3;i<a.length;i++){
            for(int j=3;j<i;j++){
                a[i][j]=a[i-1][j-1]+a[i-1][j];//看图理解
            }
        }//将一行的数据加起来
        int sum=0;
        for(int i=2;i<a[n].length;i++){
            sum += a[n][i];
        }
        System.out.println(sum);
    }
}