Iterables and Iterators

  • + 1 comment
    import itertools
    
    N, cases, K = int(input()), input().split(), int(input())
    prob_consist_a = len([combo for combo in itertools.combinations(cases, K) if 'a' in combo]) / len(list(itertools.combinations(cases, K)))
    print(prob_consist_a)