Solution for Double Square Problem in Facebook Hacker Cup Qualification Round

double_square.py

import sys
from math import sqrt

def main(filename):
    inputf = open(filename,'rU')
    totalLines = inputf.readlines()
    totalLines.pop(0)
    
    for i in totalLines:
        count = 0
        valuesArr = []
        for x in range(int(sqrt(int(i)))+1):
            y = sqrt(int(i) - x*x)
            if y == int(y):
                if x*x == y:
                    count += 2
                else:
                    count += 1
        print str(count/2)

    inputf.close()                    
    

if __name__ == '__main__':
    main(sys.argv[1])