Double Squre

double_squre.py

import math

def isSuperSqure(n):
	hlf=int(math.sqrt(n))
	start=0
	count=0
	while(hlf>=start):
		if((start+hlf)<=n):
			no=(start**2)+(hlf**2)
			if(no>n):
				hlf-=1
			elif(no
				start+=1
			else:
				count+=1
				print hlf,start,count
				hlf-=1
				start+=1
	return count
	

if __name__ == "__main__":
	f=open("double_squares.txt",'r')
	limit=int(f.readline())
	fr=open("out1.txt",'w')
	while(limit>0):
		n=int(f.readline())
		limit-=1
		fr.write(str(isSuperSqure(n))+'\n')
	fr.close()
	f.close()

Facebook Hackers-cup Double square solution