aboutsummaryrefslogtreecommitdiff
path: root/challenge-244/eric-cheung/python/ch-2.py
blob: 637dcacca2265db097fd02d84736744ba40af2b1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
## Ref.
## https://stackoverflow.com/questions/1482308/how-to-get-all-subsets-of-a-set-powerset

from itertools import chain, combinations

def GetPowerSet(arrInput):
    return list(chain.from_iterable(combinations(arrInput, rLoop) for rLoop in range(1, len(arrInput) + 1)))

arrNum = [2, 1, 4]  ## Example 1

nSum = 0

for arrLoop in GetPowerSet(arrNum):
    nMax = max(arrLoop)
    nMin = min(arrLoop)
    nSum = nSum + nMax * nMax * nMin

print (nSum)