day1/exercise/pytriads.py
author Prabhu Ramachandran <prabhu@aero.iitb.ac.in>
Mon, 28 Jun 2010 00:14:02 -0500
branchscipy2010
changeset 426 7d8738ce004d
parent 380 669b72283b55
child 428 ee689f7c3122
permissions -rw-r--r--
Adding an acknowledgement slide to session 1. Minor changes to session 2.

def is_perfect_square(n):
    i = 1
    while i * i < n:
        i += 1
    return i * i == n, i

def gcd(a, b):
    if a % b == 0:
        return b
    else:
        return gcd(b, a%b)

for a in range(3, 101):
    for b in range( a+1, 101, 2):
        if gcd( a, b ) == 1:
            is_ps, c = is_perfect_square((a * a) + (b * b))
            if is_ps: print a, b, c