#include #include #include #include #include #include #include // 内部定数 #define D_ARRAY_MAX 4000000 // 最大配列数 // 内部変数 static FILE *szpFpI; // 入力 static int si1Array[D_ARRAY_MAX]; // 配列 static int siACnt; // 配列数 // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; static int siTNo; #endif // 出力 int fOut( char *pcpLine // 1行 ) { char lc1Buf[1024]; #ifdef D_TEST fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, pcpLine)) { siRes = -1; } #else printf("%s", pcpLine); #endif return 0; } // ソート関数 - int昇順 int fSortFnc( const void *pzpVal1 // 値1 , const void *pzpVal2 // 値2 ) { int *lipVal1 = (int *)pzpVal1; int *lipVal2 = (int *)pzpVal2; // int昇順 if (*lipVal1 > *lipVal2) { return 1; } else if (*lipVal1 < *lipVal2) { return -1; } return 0; } // 検索(値の重複可・範囲取得) // 戻り値:[>=0]配列番号 [-1]なし int fBSrhMK( int piVal // 値 , int *pipArray // 配列 , int piACnt // 配列数 , int piWay // 方向 [1]右側 or [-1]左側 ) { // 初期範囲 int liSNo = 0; int liENo = piACnt - 1; // 検索 int liNo = -1; while (1) { // 中間位置 int liMNo = (liSNo + liENo) / 2; // 比較 if (piVal == pipArray[liMNo]) { // 一致 liNo = liMNo; // 次の方向 if (piWay < 0) { // 左側へ liENo = liMNo - 1; } else { // 右側へ liSNo = liMNo + 1; } } else { // 不一致 if (piVal < pipArray[liMNo]) { // 左側へ liENo = liMNo - 1; } else { // 右側へ liSNo = liMNo + 1; } } // 範囲チェック if (liSNo > liENo) { return liNo; } } return -1; } int fBSrhM( int piVal // 値 , int *pipArray // 配列 , int piACnt // 配列数 , int *pipPNo // [>=0]開始 [-1]なし , int *pipNNo // [>=0]終了 [-1]なし ) { // 左側へ *pipPNo = fBSrhMK(piVal, pipArray, piACnt, -1); if (*pipPNo < 0) { *pipNNo = -1; return -1; } // 右側へ *pipNNo = fBSrhMK(piVal, pipArray, piACnt, 1); return 0; } // 実行メイン long long fMain( ) { int i, j; char lc1Buf[1024]; // N, D - 取得 int liN, liD; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &liN, &liD); // wの2乗 + D - zの2乗 siACnt = 0; for (i = 1; i <= liN; i++) { for (j = 1; j <= liN; j++) { si1Array[siACnt] = i * i + liD - j * j; siACnt++; } } // 配列 - ソート qsort(si1Array, siACnt, sizeof(int), fSortFnc); // 個数 - 取得 long long llCnt = 0; for (i = 1; i <= liN; i++) { for (j = 1; j <= liN; j++) { int liXY = i * i + j * j; // 検索 int liPNo, liNNo; fBSrhM(liXY, si1Array, siACnt, &liPNo, &liNNo); if (liPNo >= 0) { llCnt += liNNo - liPNo + 1; } } } return llCnt; } // 1回実行 int fOne( ) { long long llRet; char lc1Buf[1024]; // 入力 - セット #ifdef D_TEST sprintf(lc1Buf, ".\\Test\\T%d.txt", siTNo); szpFpI = fopen(lc1Buf, "r"); sprintf(lc1Buf, ".\\Test\\A%d.txt", siTNo); szpFpA = fopen(lc1Buf, "r"); siRes = 0; #else szpFpI = stdin; #endif // 実行メイン llRet = fMain(); // 出力 sprintf(lc1Buf, "%lld\n", llRet); fOut(lc1Buf); // 残データ有無 #ifdef D_TEST lc1Buf[0] = '\0'; fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, "")) { siRes = -1; } #endif // テストファイルクローズ #ifdef D_TEST fclose(szpFpI); fclose(szpFpA); #endif // テスト結果 #ifdef D_TEST if (siRes == 0) { printf("OK %d\n", siTNo); } else { printf("NG %d\n", siTNo); } #endif return 0; } // プログラム開始 int main() { #ifdef D_TEST int i; for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) { siTNo = i; fOne(); } #else fOne(); #endif return 0; }