笔试题算法系列（九）百度2017不等式数列

[编程题] 不等式数列

时间限制：1秒

空间限制：32768K

度度熊最近对全排列特别感兴趣,对于1到n的一个排列,度度熊发现可以在中间根据大小关系插入合适的大于和小于符号(即 '>' 和 '<' )使其成为一个合法的不等式数列。但是现在度度熊手中只有k个小于符号即('<'')和n-k-1个大于符号(即'>'),度度熊想知道对于1至n任意的排列中有多少个排列可以使用这些符号使其为合法的不等式数列。

输入描述:

输入包括一行,包含两个整数n和k(k < n ≤ 1000)

输出描述:

输出满足条件的排列数,答案对2017取模。

输入例子1:

5 2

输出例子1:

66

代码如下：

import java.lang.Math;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Scanner;

public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String[] line;
        while(sc.hasNext()){
            line = sc.nextLine().split(" ");
            int n = Integer.parseInt(line[0]);
            int k = Integer.parseInt(line[1]);
            int[][] a=new int[n+1][n+1];
            for(int i=1;i<=n;i++){
                a[i][0]=1;
                a[i][i-1]=1;
            }
            for(int i=2;i<=n;i++){
                for(int j=1;j<=k&&j<i;j++){
                    a[i][j]=(i-j)*a[i-1][j-1]+(j+1)*a[i-1][j];
                    a[i][j]%=2017;
                }
            }
            System.out.println(a[n][k]);
        }
        sc.close();
    }
}