view python/example/array_sum.py @ 859:b8510769d001

py3 compat
author Jeff Hammel <k0scist@gmail.com>
date Wed, 11 Apr 2018 12:28:37 -0700
parents 5ba219864529
children
line wrap: on
line source

#!/usr/bin/env python

"""
given a `value` and an array,
if the sum of two numbers in the array
total the value, return `True`
(exit 0):

 ./array_sum.py 3 1 2 4  # YES!  because 1+2=3

 ./array_sum.py 4 1 2 5 6  # NO!
"""

import argparse
import sys


def sum_of_two(value, *array):

    start = 0
    end = len(array) - 1
    complements = []
    while start < end:
        start_val = array[start]
        end_val = array[end]
        _sum = start_val + end_val
        if _sum == value:
            complements.append((start_val, end_val))
            end -= 1
            start +=1
            continue
        if _sum > value:
            end -= 1
        else:
            start += 1
    return complements


def main(args=sys.argv[1:]):

    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('value', type=int)
    parser.add_argument('array', type=int, nargs='+')
    options = parser.parse_args(args)

    retval = sum_of_two(options.value, *options.array)

    print (retval)

    sys.exit(0 if retval else 1)


if __name__ == '__main__':
    main()