百度2017春季招聘笔试题

因为题目不能复制，所以这只是我对题目的二次创作，希望不会有太大的歧义

题目：输入一组数据n,k
n表示序列个数，例如n=5，则序列由1,2,3,4,5组成
k表示小于符号的数量，大于符号数量为n-k-1

现在要求根据n,k找出合法表达式的个数,结果要对2017取余
例如：输入 3,1
则可以有：
1< 3 >2,
2< 3 >1,
2>1<3,
3>1<2

故最终输出结果为4

#coding=utf-8
import sys
import copy
import itertools

if __name__ == "__main__":
    # 读取第一行的n
    data = map(int,sys.stdin.readline().strip().split())

    num = [i for i in range(1,data[0]+1)]
    nums = itertools.permutations(num)


    count = 0
    for x in nums:
        smaller = 0
        for i in range(len(x)-1):

            if(x[i]<x[i+1]):
                smaller += 1

        if smaller == data[1]:
            count += 1
    print(count%2017)