python 牛客真题数列还原

最新推荐文章于 2023-03-31 15:04:55 发布

Neekity

最新推荐文章于 2023-03-31 15:04:55 发布

阅读量224

点赞数

分类专栏： python 牛客文章标签： python 牛客

本文链接：https://blog.csdn.net/Neekity/article/details/85402097

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

牛客

19 篇文章 0 订阅

订阅专栏

考察全排列？

nk=list(input().split())
n=int(nk[0])
k=int(nk[1])
num=[int(c) for c in list(input().split())]
m=[0]*(n+1)
index=[]
for i in range(n):
    if num[i]==0: index.append(i)
    else: m[num[i]]=1
a=[]
for i in range(1,n+1):
    if m[i]==0: a.append(i)
def cal(num):
    count=0
    for i in range(n-1):
        for j in range(i+1,n):
            if num[i]<num[j]: count+=1 
    return count
def permute(a):
        res=[]
        def dfs(nums,tmp):
            if len(nums) == len(tmp):
                res.append(tmp[:])
            for m in nums:
                if m in tmp:
                    continue 
                tmp.append(m)
                dfs(nums,tmp)
                tmp.remove(m)
        dfs(a,[])
        return res
res=permute(a)
cnt=0
for list1 in res:
    count=0
    for i in range(len(a)):
        num[index[i]]=list1[i]
    if k==cal(num): cnt+=1
print(cnt)