Facebook Hacker Cup Double Squares Solution Code Program


Double Squares
A double-square number is an integer X which can be expressed as the sum of two perfect squares. For example, 10 is a double-square because 10 = 32+ 12. Your task in this problem is, given X, determine the number of ways in which it can be written as the sum of two squares. For example, 10 can only be written as 32 + 12 (we don’t count 12 + 32 as being different). On the other hand, 25 can be written as 52 + 02 or as 42 + 32.

Input

You should first read an integer N, the number of test cases. The next N lines will contain N values of X.

Constraints

0 ≤ X ≤ 2147483647
1 ≤ N ≤ 100

Output

For each value of X, you should output the number of ways to write X as the sum of two squares.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#include<iostream>
#include<fstream>
#include<algorithm>
#include<cmath>
#define SIZE 46342
using namespace std;

long long num[SIZE];

void generate()
{
     for(long long int i=0;i<SIZE;i++)
     num[i] = (i)*(i);
}

int main()
{
    generate();
    int N;
    cin>>N;
    while(N--)
    {
              long long k;
              cin>>k;
              long long int count = 0;
              for(int i=0;i<SIZE;i++)
              {
                      long long diff = k - num[i];

                      if(diff>=0)
                      {
                      if(binary_search(num,num+SIZE,diff))
                      count++;
                      }
              }
              if(count%2==0)
              cout<<count/2<<endl;
              else
              cout<<count/2+1<<endl;
    }
    return 0;
}

Source: http://www.ideone.com/TNuEz


Leave a Reply

Your email address will not be published. Required fields are marked *