Skip to content

Instantly share code, notes, and snippets.

@mym0404
Created May 17, 2022 08:57
Show Gist options
  • Save mym0404/7cf58798973e8aa5777b805a7f193e39 to your computer and use it in GitHub Desktop.
Save mym0404/7cf58798973e8aa5777b805a7f193e39 to your computer and use it in GitHub Desktop.
import itertools
n = int(input())
k = int(input())
arr = [int(input()) for _ in range(n)]
idx = [0 for _ in range(n)]
for i in range(n):
idx[i] = i
answer_set = set()
for on in itertools.permutations(idx):
idx_rev = [0 for _ in range(n)]
for i in on:
idx_rev[on[i]] = i
current = ''
for i in range(k):
current += str(arr[idx_rev[i]])
answer_set.add(current)
print(len(answer_set))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment