CSES - Putka Open 2020 – 4/5 - Neliöt
  • Time limit: 1.00 s
  • Memory limit: 512 MB

Tehtäväsi on tutkia annetusta kokonaisluvusta n, voiko sen esittää muodossa a^2+b^2, missä a ja b ovat kokonaislukuja.

Esimerkiksi kun n=89, voidaan valita a=5 ja b=8, jolloin 5^2+8^2=89.

Syöte

Syötteen ensimmäisellä rivillä on kokonaisluku t: testien määrä.

Tämän jälkeen syötteessä on t riviä, joista jokaisella on kokonaisluku n.

Tuloste

Tulosta jokaisesta testistä YES, jos luvun n voi esittää muodossa a^2+b^2, ja muuten NO.

Esimerkki

Syöte:

5
2
6
9
89
91

Tuloste:

YES
NO
YES
YES
NO

Osatehtävä 1 (28 pistettä)

  • 1 \le t \le 100
  • 1 \le n \le 1000

Osatehtävä 2 (72 pistettä)

  • 1 \le t \le 100
  • 1 \le n \le 10^9