結果

問題 No.800 四平方定理
ユーザー asugen0402asugen0402
提出日時 2019-05-15 13:47:48
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 1,311 ms / 2,000 ms
コード長 3,987 bytes
コンパイル時間 290 ms
コンパイル使用メモリ 33,024 KB
実行使用メモリ 30,384 KB
最終ジャッジ日時 2024-09-13 23:26:58
合計ジャッジ時間 17,564 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
6,812 KB
testcase_01 AC 15 ms
6,940 KB
testcase_02 AC 12 ms
6,940 KB
testcase_03 AC 12 ms
6,940 KB
testcase_04 AC 11 ms
6,940 KB
testcase_05 AC 12 ms
6,944 KB
testcase_06 AC 18 ms
6,944 KB
testcase_07 AC 13 ms
6,944 KB
testcase_08 AC 11 ms
6,944 KB
testcase_09 AC 17 ms
6,944 KB
testcase_10 AC 590 ms
16,256 KB
testcase_11 AC 595 ms
18,556 KB
testcase_12 AC 628 ms
18,624 KB
testcase_13 AC 567 ms
19,008 KB
testcase_14 AC 574 ms
17,324 KB
testcase_15 AC 643 ms
19,756 KB
testcase_16 AC 578 ms
18,340 KB
testcase_17 AC 599 ms
18,792 KB
testcase_18 AC 666 ms
18,560 KB
testcase_19 AC 648 ms
19,108 KB
testcase_20 AC 1 ms
6,944 KB
testcase_21 AC 1 ms
6,940 KB
testcase_22 AC 720 ms
19,160 KB
testcase_23 AC 1,311 ms
30,116 KB
testcase_24 AC 1,094 ms
29,420 KB
testcase_25 AC 1,297 ms
29,128 KB
testcase_26 AC 1 ms
6,940 KB
testcase_27 AC 1 ms
6,944 KB
testcase_28 AC 1,049 ms
29,860 KB
testcase_29 AC 1,135 ms
30,384 KB
testcase_30 AC 1,103 ms
28,732 KB
testcase_31 AC 968 ms
27,816 KB
testcase_32 AC 1,145 ms
29,548 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <float.h>
#include <limits.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

// 内部定数
#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				// <I> １行
)
{
	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			// <I> 値１
	, const void *pzpVal2		// <I> 値２
)
{
	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					// <I> 値
	, int *pipArray				// <I> 配列
	, int piACnt				// <I> 配列数
	, int piWay					// <I> 方向 [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					// <I> 値
	, int *pipArray				// <I> 配列
	, int piACnt				// <I> 配列数
	, int *pipPNo				// <O> [>=0]開始 [-1]なし
	, int *pipNNo				// <O> [>=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;
}

// １回実行
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;
}
0