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

Tehtäväsi on tutkia annetusta kokonaisluvusta nn, voiko sen esittää muodossa a2+b2a^2+b^2, missä aa ja bb ovat kokonaislukuja.

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

Syöte

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

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

Tuloste

Tulosta jokaisesta testistä YES, jos luvun nn voi esittää muodossa a2+b2a^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ä)

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

Osatehtävä 2 (72 pistettä)

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