We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Python
  3. Itertools
  4. Iterables and Iterators
  5. Discussions

Iterables and Iterators

  • kcmutya
     + 0 comments
    # Enter your code here. Read input from STDIN. Print output to STDOUT
import itertools

N = int(input())
cases = list(input().split())
K = int(input())

all_combinations = list(itertools.combinations(cases, K))

valid_combinations = [combo for combo in all_combinations if 'a' in combo]

prob_consist_a = len(valid_combinations) / len(all_combinations)
print(prob_consist_a)