python_training/practices/interview03_solution.py at main · DevCTx/python_training · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from itertools import combinations

def val_in_sum_list(N,numlist):
    '''
    Returns all list of numbers from numlist whose sum is equal to N

    >>> val_in_sum_list(5,[1,2,3])
    [(2, 3)]
    >>> val_in_sum_list(5,[1,2,3,3,4,6,7])
    [(1, 4), (2, 3)]
    >>> val_in_sum_list(17,[1,2,3,3,4,6,7])
    [(1, 2, 3, 4, 7), (1, 3, 3, 4, 6), (1, 3, 6, 7), (3, 3, 4, 7), (4, 6, 7)]
    >>> val_in_sum_list(23,[1,2,3,3,4,6,7])
    [(1, 2, 3, 4, 6, 7), (3, 3, 4, 6, 7)]
    >>> val_in_sum_list(27,[1,2,3,3,4,6,7])
    []
    '''
    l = [t for i in range(N) for t in combinations(numlist, i) if sum(t) == N ]

    return sorted(list(set(l)))