From d99416a28a49cd9069d61476f2950e400b5523dc Mon Sep 17 00:00:00 2001 From: Gea-Suan Lin Date: Thu, 17 Nov 2022 09:26:05 +0800 Subject: [PATCH] Add a utility to list all cost combinations. --- combination.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100755 combination.py diff --git a/combination.py b/combination.py new file mode 100755 index 0000000..97abe2c --- /dev/null +++ b/combination.py @@ -0,0 +1,16 @@ +#!/usr/bin/env python3 + +import itertools +import sys + +class Main(): + def main(self): + costs = list(map(lambda x: int(x), sys.argv[1:])) + + for cnt in range(1, len(costs) + 1): + for item in itertools.combinations(costs, cnt): + cost = sum(item) + print('{}: {}'.format(cost, list(item))) + +if __name__ == '__main__': + Main().main()