版权声明:本文为博主原创文章,未经博主允许不得转载。转载请注明作者和出处:https://blog.csdn.net/weixin_41481113 https://blog.csdn.net/weixin_41481113/article/details/83927684
题目描述
牛牛的作业薄上有一个长度为 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)