Prime Judge

时间限制
1000 ms 内存限制 65536 KB

题目描述

众所周知，如果一个正整数只能被1和自身整除，那么该数被称为素数。题目的任务很简单，就是判定一个数是否是一个素数。 只不过可能数的形式与正整数有一些不同，数的形式为a+bi，其中a、b为整数，且ii被定义为-1。如果a+bi能被分解为(a1+b1i)(a2+b2i)的形式，那么该数不是素数；否则，该数是素数。其中a1 、b1、 a2 、b2均为整数，且1，-1，0，i，-i不能作为被分解的因子。 注意，1，-1，0，i，-i均不为素数。

输入格式

输入包括若干组数据，每行包含2个数a、b，表示一个形如a+bi的数。a，b小于10000。

输出格式

对应于输入每行，如果输入的数为素数，则输出“YES”，否则输出“NO”。（不包括引号）

输入样例

-10 2
3 0

输出样例

NO
YES

根据复数的性质可知：a^2+b^2=(a1^2+b1^2)(a2^2+b2^2)。

算算复杂度，由于a,b<10000，令z1=a1^2+b1^2,z2=a2^2+b2^2，枚举z1 ，范围为sqrt（10000^2*2) （大概14320左右）得到z2,再枚举a2^2得到b2^2，在判断b2^2是不是平方数，如果是，则输出NO，否则输出YES。

我事先打了平方和表（14320之内）

代码如下：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 10000
using namespace std;
long long  a,b,a1,a2,b1,b2;
long long  squ[]={2,4,5,8,9,10,13,16,17,18,20,25,26,29,32,34,36,37,40,41,45,49,50,52,53,58,61,64,65,68,72,73,74,80,81,82,85,89,90,97,98,100,101,104,106,109,113,116,117,121,122,125,128,130,136,137,144,145,146,148,149,153,157,160,162,164,169,170,173,178,180,181,185,193,194,196,197,200,202,205,208,212,218,221,225,226,229,232,233,234,241,242,244,245,250,256,257,260,261,265,269,272,274,277,281,288,289,290,292,293,296,298,305,306,313,314,317,320,324,325,328,333,337,338,340,346,349,353,356,360,361,362,365,369,370,373,377,386,388,389,392,394,397,400,401,404,405,409,410,416,421,424,425,433,436,441,442,445,449,450,452,457,458,461,464,466,468,477,481,482,484,485,488,490,493,500,505,509,512,514,520,521,522,529,530,533,538,541,544,545,548,549,554,557,562,565,569,576,577,578,580,584,585,586,592,593,596,601,605,610,612,613,617,625,626,628,629,634,637,640,641,648,650,653,656,657,661,666,673,674,676,677,680,685,689,692,697,698,701,706,709,712,720,722,724,725,729,730,733,738,740,745,746,754,757,761,765,769,772,773,776,778,784,785,788,793,794,797,800,801,802,808,809,810,818,820,821,829,832,833,841,842,845,848,850,853,857,865,866,872,873,877,881,882,884,890,898,900,901,904,905,909,914,916,922,925,928,929,932,936,937,941,949,953,954,961,962,964,965,968,970,976,977,980,981,985,986,997,1000,1009,1010,1013,1017,1018,1021,1024,1025,1028,1033,1037,1040,1042,1044,1049,1053,1058,1060,1061,1066,1069,1073,1076,1082,1088,1089,1090,1093,1096,1097,1098,1105,1108,1109,1114,1117,1124,1125,1129,1130,1138,1145,1152,1153,1154,1156,1157,1160,1165,1168,1170,1172,1181,1184,1186,1189,1192,1193,1201,1202,1205,1210,1213,1217,1220,1224,1225,1226,1229,1233,1234,1237,1241,1249,1250,1252,1256,1258,1261,1268,1274,1277,1280,1282,1285,1289,1296,1297,1300,1301,1305,1306,1312,1313,1314,1321,1322,1325,1332,1341,1345,1346,1348,1352,1354,1360,1361,1369,1370,1373,1377,1378,1381,1384,1385,1394,1396,1402,1405,1409,1412,1413,1417,1418,1421,1424,1429,1433,1440,1444,1445,1448,1450,1453,1458,1460,1465,1466,1469,1476,1480,1481,1489,1490,1492,1493,1508,1513,1514,1517,1521,1522,1525,1530,1537,1538,1544,1546,1549,1552,1553,1556,1557,1565,1568,1570,1573,1576,1585,1586,1588,1594,1597,1600,1601,1602,1604,1609,1613,1616,1618,1620,1621,1625,1629,1636,1637,1640,1642,1649,1657,1658,1664,1665,1666,1669,1681,1682,1684,1685,1690,1693,1696,1697,1700,1706,1709,1714,1717,1721,1730,1732,1733,1737,1741,1744,1745,1746,1753,1754,1762,1764,1765,1768,1769,1773,1777,1780,1781,1789,1796,1800,1801,1802,1805,1808,1810,1813,1818,1825,1828,1832,1844,1845,1849,1850,1853,1856,1858,1861,1864,1865,1872,1873,1874,1877,1882,1885,1889,1898,1901,1906,1908,1913,1921,1922,1924,1928,1930,1933,1936,1937,1940,1945,1949,1952,1954,1960,1961,1962,1970,1972,1973,1985,1989,1993,1994,1997,2000,2005,2009,2017,2018,2020,2025,2026,2029,2034,2036,2041,2042,2045,2048,2050,2053,2056,2057,2061,2066,2069,2074,2080,2081,2084,2088,2089,2097,2098,2105,2106,2113,2116,2117,2120,2122,2125,2129,2132,2137,2138,2141,2146,2152,2153,2161,2164,2165,2169,2173,2176,2178,2180,2186,2192,2194,2196,2197,2205,2209,2210,2213,2216,2218,2221,2225,2228,2234,2237,2245,2248,2249,2250,2257,2258,2260,2269,2273,2276,2281,2285,2290,2293,2297,2304,2305,2306,2308,2309,2312,2313,2314,2320,2329,2330,2333,2336,2340,2341,2344,2349,2353,2357,2362,2368,2372,2377,2378,2381,2384,2385,2386,2389,2393,2401,2402,2404,2405,2410,2417,2420,2421,2425,2426,2434,2437,2440,2441,2448,2450,2452,2458,2465,2466,2468,2473,2474,2477,2482,2493,2498,2500,2501,2504,2509,2512,2516,2521,2522,2525,2529,2533,2536,2545,2548,2549,2554,2557,2560,2561,2564,2570,2578,2581,2592,2593,2594,2597,2600,2601,2602,2605,2609,2610,2612,2617,2621,2624,2626,2628,2633,2637,2642,2644,2645,2650,2657,2664,2665,2669,2677,2682,2689,2690,2692,2693,2696,2701,2704,2705,2708,2713,2720,2722,2725,2729,2738,2740,2741,2745,2746,2749,2753,2754,2756,2762,2768,2770,2777,2785,2788,2789,2792,2797,2801,2804,2809,2810,2813,2817,2818,2824,2825,2826,2833,2834,2836,2837,2842,2845,2848,2853,2857,2858,2861,2866,2873,2880,2885,2888,2890,2896,2897,2900,2906,2909,2916,2917,2920,2925,2929,2930,2932,2938,2941,2952,2953,2957,2960,2962,2965,2969,2977,2978,2980,2984,2986,2989,2993,2997,3001,3005,3016,3025,3026,3028,3029,3033,3034,3037,3041,3042,3044,3049,3050,3060,3061,3065,3074,3076,3077,3085,3088,3089,3092,3098,3104,3106,3109,3112,3114,3121,3125,3130,3133,3136,3137,3140,3141,3145,3146,3152,3161,3169,3170,3172,3176,3177,3181,3185,3188,3194,3200,3202,3204,3205,3208,3209,3217,3218,3221,3226,3229,3232,3233,3236,3240,3242,3249,3250,3253,3257,3258,3265,3272,3274,3277,3280,3281,3284,3285,3293,3298,3301,3305,3313,3314,3316,3321,3328,3329,3330,3332,3338,3341,3349,3357,3361,3362,3364,3365,3368,3370,3373,3380,3385,3386,3389,3392,3393,3394,3400,3412,3413,3418,3425,3428,3433,3434,3442,3445,3449,3457,3460,3461,3464,3466,3469,3474,3481,3482,3485,3488,3490,3492,3497,3501,3505,3506,3508,3509,3517,3524,3528,3529,3530,3533,3536,3538,3541,3545,3546,3554,3557,3560,3562,3573,3577,3578,3581,3589,3592,3593,3600,3601,3602,3604,3609,3610,3613,3616,3617,3620,3625,3626,3636,3637,3645,3649,3650,3653,3656,3664,3665,3673,3677,3681,3688,3690,3697,3698,3700,3701,3706,3709,3712,3716,3721,3722,3725,3728,3730,3733,3737,3744,3746,3748,3754,3757,3761,3764,3769,3770,3778,3785,3789,3793,3796,3797,3802,3805,3809,3812,3816,3821,3825,3826,3833,3842,3844,3845,3848,3853,3856,3860,3865,3866,3869,3872,3874,3877,3880,3881,3889,3890,3893,3897,3898,3904,3908,3917,3920,3922,3924,3925,3929,3940,3944,3946,3961,3965,3969,3970,3973,3977,3978,3985,3986,3988,3989,3994,4000,4001,4005,4010,4013,4018,4021,4033,4034,4036,4040,4041,4045,4049,4050,4052,4057,4058,4068,4069,4072,4073,4082,4084,4090,4093,4096,4097,4100,4105,4106,4112,4113,4114,4121,4122,4129,4132,4133,4138,4141,4145,4148,4149,4153,4157,4160,4162,4165,4168,4176,4177,4178,4181,4194,4196,4201,4205,4210,4212,4217,4225,4226,4229,4232,4234,4240,4241,4244,4250,4253,4258,4261,4264,4265,4273,4274,4276,4282,4285,4289,4292,4293,4297,4304,4306,4321,4322,4325,4328,4329,4330,4337,4338,4346,4349,4352,4356,4357,4360,4361,4365,4369,4372,4373,4381,4384,4385,4388,4392,4394,4397,4405,4409,4410,4418,4420,4421,4426,4432,4436,4437,4441,4442,4450,4453,4456,4457,4468,4469,4474,4477,4481,4489,4490,4493,4496,4498,4500,4505,4513,4514,4516,4517,4520,4525,4537,4538,4545,4546,4549,4552,4553,4561,4562,4570,4573,4580,4581,4586,4589,4594,4597,4608,4610,4612,4616,4618,4621,4624,4625,4626,4628,4633,4637,4640,4645,4649,4657,4658,4660,4666,4672,4673,4680,4682,4685,4688,4689,4693,4698,4705,4706,4709,4714,4717,4721,4724,4729,4733,4736,4744,4745,4753,4754,4756,4761,4762,4765,4768,4770,4772,4777,4778,4786,4789,4793,4797,4801,4802,4804,4805,4808,4810,4813,4817,4820,4825,4834,4840,4842,4849,4850,4852,4861,4868,4869,4874,4877,4880,4882,4885,4889,4896,4900,4901,4904,4905,4909,4913,4916,4925,4930,4932,4933,4936,4937,4941,4946,4948,4949,4954,4957,4961,4964,4969,4973,4981,4985,4986,4993,4996,5000,5002,5008,5009,5013,5017,5018,5021,5024,5032,5041,5042,5044,5045,5050,5057,5058,5065,5066,5069,5072,5077,5081,5085,5090,5096,5098,5101,5105,5108,5113,5114,5120,5121,5122,5125,5128,5140,5141,5153,5156,5161,5162,5165,5184,5185,5186,5188,5189,5193,5194,5197,5200,5202,5204,5209,5210,5213,5218,5220,5224,5233,5234,5237,5242,5245,5248,5249,5252,5256,5261,5265,5266,5273,5274,5281,5284,5288,5290,5297,5300,5305,5309,5314,5317,5321,5328,5329,5330,5333,5337,5338,5341,5345,5353,5354,5364,5365,5378,5380,5381,5384,5386,5389,5392,5393,5402,5408,5409,5410,5413,5416,5417,5426,5429,5437,5440,5441,5444,5445,5449,5450,5458,5465,5473,5476,5477,5480,5482,5485,5490,5492,5498,5501,5506,5508,5512,5513,5517,5521,5524,5525,5536,5537,5540,5545,5553,5554,5557,5569,5570,5573,5576,5578,5581,5584,5585,5594,5597,5602,5608,5617,5618,5620,5625,5626,5629,5634,5636,5641,5645,5648,5650,5652,5653,5657,5661,5666,5668,5669,5672,5674,5684,5689,5690,5693,5696,5701,5706,5713,5714,5716,5717,5722,5725,5729,5732,5733,5737,5741,5746,5749,5760,5765,5769,5770,5776,5777,5780,5785,5792,5794,5800,5801,5809,5812,5813,5818,5821,5825,5832,5834,5837,5840,5849,5850,5857,5858,5860,5861,5864,5869,5876,5877,5881,5882,5897,5904,5905,5906,5913,5914,5917,5920,5924,5929,5930,5933,5938,5941,5945,5949,5953,5954,5956,5960,5965,5968,5972,5978,5981,5986,5989,5993,5994,6001,6002,6005,6010,6025,6029,6032,6037,6050,6052,6053,6056,6057,6058,6065,6066,6068,6073,6074,6082,6084,6085,6088,6089,6093,6098,6100,6101,6109,6113,6120,6121,6122,6125,6130,6133,6137,6145,6148,6152,6154,6161,6165,6170,6173,6176,6178,6184,6185,6196,6197,6201,6205,6208,6212,6217,6218,6221,6224,6228,6229,6241,6242,6245,6250,6253,6257,6260,6266,6269,6272,6273,6274,6277,6280,6282,6290,6292,6301,6304,6305,6309,6317,6322,6329,6337,6338,6340,6341,6344,6352,6353,6354,6361,6362,6370,6373,6376,6381,6385,6388,6389,6397,6400,6401,6404,6408,6409,6410,6413,6416,6418,6421,6425,6434,6436,6437,6442,6445,6449,6452,6458,6464,6466,6469,6472,6473,6480,6481,6484,6485,6497,6498,6500,6505,6506,6514,6516,6521,6525,6529,6530,6544,6548,6553,6554,6560,6561,6562,6565,6568,6569,6570,6577,6581,6586,6596,6597,6602,6605,6610,6613,6617,6625,6626,6628,6632,6637,6641,6642,6649,6653,6656,6658,6660,6661,6664,6673,6676,6682,6689,6697,6698,6701,6705,6709,6713,6714,6722,6724,6725,6728,6730,6733,6736,6737,6740,6746,6749,6757,6760,6761,6770,6772,6773,6778,6781,6784,6786,6788,6793,6800,6805,6813,6817,6824,6826,6829,6833,6836,6841,6845,6849,6850,6856,6857,6865,6866,6868,6869,6877,6884,6885,6889,6890,6893,6898,6905,6914,6917,6920,6921,6922,6925,6928,6929,6932,6938,6948,6949,6953,6957,6961,6962,6964,6970,6976,6977,6980,6984,6989,6994,6997,7001,7002,7010,7012,7013,7016,7018,7025,7033,7034,7045,7048,7056,7057,7058,7060,7065,7066,7069,7072,7076,7081,7082,7085,7090,7092,7093,7105,7108,7109,7114,7120,7121,7124,7129,7137,7141,7145,7146,7154,7156,7157,7162,7165,7173,7177,7178,7184,7186,7193,7200,7202,7204,7208,7209,7213,7218,7220,7225,7226,7229,7232,7234,7237,7240,7241,7250,7252,7253,7261,7265,7272,7274,7281,7289,7290,7297,7298,7300,7301,7306,7309,7312,7321,7325,7328,7330,7333,7345,7346,7349,7354,7361,7362,7369,7373,7376,7380,7381,7389,7393,7394,7396,7397,7400,7402,7405,7412,7417,7418,7421,7424,7432,7433,7442,7444,7445,7450,7453,7456,7457,7460,7461,7465,7466,7474,7477,7481,7488,7489,7492,7496,7497,7501,7508,7514,7517,7522,7528,7529,7537,7538,7540,7541,7549,7556,7561,7565,7569,7570,7573,7577,7578,7585,7586,7589,7592,7594,7604,7605,7610,7618,7621,7624,7625,7632,7633,7642,7649,7650,7652,7666,7669,7673,7677,7681,7684,7685,7688,7690,7693,7696,7706,7709,7712,7713,7717,7720,7730,7732,7738,7741,7744,7745,7748,7753,7754,7757,7760,7762,7765,7769,7778,7780,7785,7786,7789,7793,7794,7796,7801,7808,7813,7816,7817,7825,7829,7834,7837,7840,7841,7844,7848,7850,7853,7857,7858,7865,7873,7877,7880,7888,7892,7893,7897,7901,7913,7921,7922,7925,7929,7930,7933,7937,7938,7940,7946,7949,7954,7956,7957,7969,7970,7972,7976,7978,7985,7988,7993,8000,8002,8005,8009,8010,8017,8020,8021,8026,8033,8036,8042,8045,8053,8065,8066,8068,8069,8072,8077,8080,8081,8082,8089,8090,8093,8098,8100,8101,8104,8105,8109,8114,8116,8117,8125,8136,8138,8144,8145,8146,8149,8161,8164,8168,8177,8180,8181,8185,8186,8192,8194,8200,8209,8210,8212,8221,8224,8226,8228,8233,8237,8242,8244,8245,8249,8258,8264,8266,8269,8273,8276,8281,8282,8285,8290,8293,8296,8297,8298,8306,8314,8317,8320,8321,8324,8325,8329,8330,8333,8336,8345,8352,8353,8354,8356,8357,8361,8362,8369,8377,8381,8388,8389,8392,8402,8405,8410,8420,8424,8425,8429,8433,8434,8450,8452,8458,8461,8464,8465,8468,8469,8473,8477,8480,8482,8485,8488,8489,8497,8500,8501,8506,8513,8516,8521,8522,8528,8530,8537,8541,8545,8546,8548,8552,8564,8570,8573,8577,8578,8581,8584,8585,8586,8593,8594,8597,8605,8608,8609,8612,8621,8629,8633,8641,8642,8644,8649,8650,8653,8656,8658,8660,8665,8669,8674,8676,8677,8681,8685,8689,8692,8693,8698,8704,8705,8712,8713,8714,8720,8722,8725,8730,8737,8738,8741,8744,8746,8749,8753,8761,8762,8765,8768,8770,8776,8784,8788,8793,8794,8801,8810,8818,8820,8821,8825,8829,8833,8836,8837,8840,8842,8845,8849,8852,8857,8861,8864,8865,8869,8872,8874,8882,8884,8885,8893,8900,8905,8906,8912,8914,8917,8929,8933,8936,8938,8941,8945,8948,8954,8957,8962,8969,8973,8978,8980,8986,8989,8992,8993,8996,9000,9001,9005,9010,9013,9025,9026,9028,9029,9032,9034,9040,9041,9049,9050,9061,9065,9074,9076,9077,9081,9089,9090,9092,9098,9104,9106,9109,9113,9117,9122,9124,9125,9133,9137,9140,9146,9153,9157,9160,9161,9162,9169,9172,9173,9178,9181,9188,9189,9193,9194,9197,9209,9216,9217,9220,9221,9224,9225,9232,9236,9241,9242,9245,9248,9250,9252,9256,9257,9265,9266,9274,9277,9280,9281,9290,9293,9297,9298,9305,9314,9316,9320,9325,9332,9333,9337,9341,9344,9346,9349,9360,9364,9365,9370,9376,9377,9378,9385,9386,9389,9396,9397,9409,9410,9412,9413,9418,9421,9425,9428,9433,9434,9437,9441,9442,9445,9448,9457,9458,9461,9466,9469,9472,9473,9477,9488,9490,9497,9505,9506,9508,9509,9512,9521,9522,9524,9529,9530,9533,9536,9540,9544,9549,9553,9554,9556,9565,9572,9577,9578,9586,9593,9594,9601,9602,9604,9605,9608,9610,9613,9616,9620,9621,9626,9629,9634,9640,9649,9650,9653,9657,9661,9665,9668,9673,9677,9680,9684,9685,9689,9697,9698,9700,9701,9704,9721,9722,9725,9733,9736,9738,9745,9748,9749,9754,9760,9764,9769,9770,9773,9778,9781,9792,9797,9800,9801,9802,9805,9808,9809,9810,9817,9818,9826,9829,9832,9833,9837,9841,9850,9857,9860,9864,9865,9866,9872,9873,9874,9881,9882,9892,9893,9896,9898,9901,9908,9914,9922,9925,9928,9929,9938,9941,9945,9946,9949,9953,9962,9965,9970,9972,9973,9981,9985,9986,9992,9997,10000,10001,10004,10009,10016,10018,10025,10026,10034,10036,10037,10042,10045,10048,10049,10053,10057,10061,10064,10069,10081,10082,10084,10085,10088,10090,10093,10100,10114,10116,10121,10125,10130,10132,10133,10138,10141,10144,10145,10154,10161,10162,10169,10170,10177,10180,10181,10192,10193,10196,10201,10202,10205,10210,10216,10217,10225,10226,10228,10229,10237,10240,10242,10244,10249,10250,10253,10256,10265,10273,10280,10282,10285,10289,10301,10305,10306,10309,10312,10313,10321,10322,10324,10330,10333,10337,10345,10357,10361,10368,10369,10370,10372,10376,10377,10378,10386,10388,10394,10397,10400,10404,10405,10408,10413,10418,10420,10421,10426,10429,10433,10436,10440,10441,10445,10448,10453,10457,10466,10468,10469,10474,10477,10484,10485,10489,10490,10496,10498,10501,10504,10512,10513,10517,10522,10525,10529,10530,10532,10537,10546,10548,10553,10562,10565,10568,10573,10576,10580,10585,10589,10594,10597,10600,10601,10609,10610,10613,10618,10625,10628,10629,10634,10642,10645,10656,10657,10658,10660,10666,10673,10674,10676,10682,10685,10690,10693,10701,10705,10706,10708,10709,10728,10729,10730,10733,10737,10753,10756,10760,10762,10765,10768,10769,10772,10777,10778,10781,10784,10786,10789,10804,10805,10809,10816,10817,10818,10820,10825,10826,10829,10832,10834,10837,10841,10845,10852,10853,10858,10861,10865,10874,10877,10880,10882,10888,10889,10890,10897,10898,10900,10909,10916,10917,10930,10933,10937,10946,10949,10952,10953,10954,10957,10960,10961,10964,10970,10973,10980,10984,10985,10993,10996,11002,11009,11012,11016,11024,11025,11026,11029,11034,11041,11042,11045,11048,11050,11057,11061,11065,11069,11072,11074,11080,11089,11090,11093,11097,11101,11105,11106,11108,11113,11114,11117,11125,11133,11138,11140,11141,11146,11149,11152,11156,11161,11162,11168,11169,11170,11173,11177,11185,11188,11194,11197,11204,11213,11216,11221,11225,11234,11236,11237,11240,11241,11245,11250,11252,11257,11258,11261,11268,11272,11273,11281,11282,11285,11290,11296,11300,11304,11306,11314,11317,11321,11322,11329,11332,11336,11338,11344,11345,11348,11349,11353,11357,11365,11368,11369,11378,11380,11386,11392,11393,11401,11402,11405,11412,11413,11417,11425,11426,11428,11432,11434,11437,11441,11444,11449,11450,11453,11458,11461,11464,11465,11466,11474,11482,11485,11489,11492,11493,11497,11498,11509,11513,11520,11521,11525,11530,11538,11540,11545,11549,11552,11554,11560,11565,11570,11581,11584,11588,11593,11597,11600,11601,11602,11617,11618,11621,11624,11626,11629,11633,11636,11642,11645,11650,11657,11664,11665,11668,11673,11674,11677,11680,11681,11689,11698,11700,11701,11705,11709,11713,11714,11716,11717,11720,11722,11728,11729,11737,11738,11745,11752,11754,11762,11764,11765,11773,11777,11785,11789,11794,11801,11808,11809,11810,11812,11813,11817,11821,11826,11828,11833,11834,11840,11848,11849,11858,11860,11861,11866,11876,11881,11882,11885,11889,11890,11897,11898,11905,11906,11908,11909,11912,11917,11920,11925,11930,11933,11936,11941,11944,11945,11953,11956,11962,11965,11969,11972,11978,11981,11986,11988,12002,12004,12005,12010,12013,12017,12020,12025,12037,12041,12049,12050,12053,12058,12064,12069,12073,12074,12077,12085,12097,12100,12101,12104,12105,12106,12109,12112,12113,12114,12116,12125,12130,12132,12136,12137,12146,12148,12149,12157,12161,12164,12168,12170,12176,12178,12181,12185,12186,12193,12196,12197,12200,12202,12205,12209,12218,12221,12226,12233,12240,12241,12242,12244,12249,12250,12253,12260,12266,12269,12274,12277,12281,12289,12290,12296,12301,12304,12308,12317,12321,12322,12325,12329,12330,12337,12340,12346,12349,12352,12356,12357,12365,12368,12370,12373,12377,12385,12389,12392,12393,12394,12401,12402,12409,12410,12413,12416,12421,12424,12429,12433,12434,12436,12437,12442,12448,12456,12457,12458,12461,12465,12469,12473,12482,12484,12490,12493,12497,12500,12505,12506,12514,12517,12520,12532,12538,12541,12544,12545,12546,12548,12553,12554,12557,12560,12564,12569,12577,12580,12584,12589,12593,12601,12602,12605,12608,12610,12613,12618,12625,12629,12634,12637,12641,12644,12645,12653,12658,12665,12674,12676,12680,12681,12682,12688,12689,12697,12701,12704,12706,12708,12713,12717,12721,12722,12724,12725,12740,12745,12746,12752,12753,12757,12762,12769,12770,12773,12776,12778,12781,12785,12789,12794,12800,12802,12805,12808,12809,12816,12818,12820,12821,12826,12829,12832,12833,12836,12841,12842,12850,12853,12861,12868,12869,12872,12874,12884,12889,12890,12893,12897,12898,12904,12905,12913,12916,12917,12928,12932,12937,12938,12941,12944,12946,12953,12960,12961,12962,12965,12968,12970,12973,12985,12994,12996,12997,13000,13001,13005,13009,13010,13012,13021,13025,13028,13032,13033,13037,13042,13045,13049,13050,13058,13060,13061,13073,13077,13085,13088,13093,13096,13105,13106,13108,13109,13117,13120,13121,13122,13124,13130,13136,13138,13140,13141,13154,13162,13165,13169,13172,13177,13181,13185,13189,13192,13194,13204,13210,13213,13217,13220,13221,13225,13226,13229,13234,13241,13249,13250,13252,13253,13256,13261,13264,13273,13274,13282,13284,13285,13289,13297,13298,13306,13309,13312,13313,13316,13320,13322,13325,13328,13329,13337,13345,13346,13352,13357,13364,13369,13378,13381,13385,13394,13396,13397,13401,13402,13410,13417,13418,13421,13426,13428,13429,13437,13441,13444,13445,13448,13450,13456,13457,13460,13465,13466,13469,13472,13474,13477,13480,13481,13492,13498,13505,13513,13514,13520,13522,13525,13537,13540,13544,13546,13549,13553,13556,13562,13565,13568,13572,13573,13576,13577,13586,13597,13600,13610,13613,13617,13621,13625,13626,13633,13634,13637,13645,13648,13649,13652,13653,13658,13666,13669,13672,13673,13681,13682,13689,13690,13693,13697,13698,13700,13705,13709,13712,13714,13721,13725,13729,13730,13732,13736,13738,13745,13753,13754,13757,13765,13768,13769,13770,13778,13780,13781,13786,13789,13793,13796,13801,13810,13817,13828,13829,13833,13834,13837,13840,13841,13842,13844,13850,13856,13858,13864,13873,13876,13877,13885,13896,13897,13898,13901,13906,13913,13914,13921,13922,13924,13925,13928,13933,13940,13941,13945,13949,13952,13954,13957,13960,13968,13969,13973,13977,13978,13985,13988,13994,13997,14002,14004,14005,14009,14013,14020,14024,14026,14029,14032,14033,14036,14045,14050,14057,14065,14066,14068,14081,14085,14089,14090,14093,14096,14112,14114,14116,14120,14125,14130,14132,14138};
long long  size_squ=4000;
bool judge=false;

int main()
{
    long long  sq;
    long long  size_z2=sqrt(N*N*2)/2;
    while(~scanf("%lld%lld",&a,&b))
    {
        if((a*a+b*b==0)||(a*a+b*b==1))
        {
            printf("NO\n");
            continue;
        }
        judge=false;
        sq=a*a+b*b;
        long long z2;
        for(long long  i=0;i<size_squ&&sq>=squ[i];i++)
        {
            b2=0;
            double temp=double(sq)/double(squ[i]);
            if(temp!=int (temp))
                continue;
            else
                z2=int(temp);
            for(long long  j=0;j<size_z2&&b2>=0&&z2!=1;j++)
            {
                b2=z2-j*j;
                int aa=sqrt(b2);
                if(b2==aa*aa)
                {
                    judge=true;
                    break;
                }
            }
            if(judge) break;
        }
        if(judge)
            printf("NO\n");
        else
            printf("YES\n");
    }
    return 0;
}

之前忘记处理特值（0,1,-1）被坑了一晚上 T_T 说起来都是泪啊！

丁神的数论法也是好牛，本人数论太菜看不懂。。。

7.12归来赛_B