牛客网在线编程（15）：数列还原-python

题目描述

牛牛的作业薄上有一个长度为 n 的排列 A，这个排列包含了从1到n的n个数，但是因为一些原因，其中有一些位置（不超过 10 个）看不清了，但是牛牛记得这个数列顺序对的数量是 k，顺序对是指满足 i < j 且 A[i] < A[j] 的对数，请帮助牛牛计算出，符合这个要求的合法排列的数目。

输入描述:

每个输入包含一个测试用例。每个测试用例的第一行包含两个整数 n 和 k（1 <= n <= 100, 0 <= k <= 1000000000），接下来的 1 行，包含 n 个数字表示排列 A，其中等于0的项表示看不清的位置（不超过 10 个）。

输出描述:

输出一行表示合法的排列数目。

示例1

输入

5 5
4 0 0 2 0

输出

2

代码实现 

 

from itertools import permutations

n,k=map(int,input().split())
A=list(map(int,input().split()))

result=0
k_=0
pos=[i for i in range(n) if A[i]==0]
queshi=[i for i in range(1,n+1) if i not in A]

for i in range(1,n):
    if i not in pos:
        for j in range(i):
            if 0<A[j]<A[i]:
                k_+=1

for lis in permutations(queshi):
    K=0
    for i,num in enumerate(lis):
        for j in range(pos[i]):
            if 0<A[j]<num:
                K+=1
            elif A[j]==0:
                if lis[pos.index(j)]<num:
                    K+=1
        for j in range(pos[i],n):
            if num<A[j]:
                   K+=1
    if K+k_==k:
        result+=1
print(result)