結果

問題 No.412 花火大会
ユーザー asugen0402
提出日時 2019-04-04 17:15:07
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 4,860 bytes
コンパイル時間 229 ms
コンパイル使用メモリ 33,152 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-26 00:20:24
合計ジャッジ時間 1,289 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

// 内部定数
#define D_FCNT			3										// 家族数
#define D_COMB_MAX		35										// 最大組み合わせ表数
#define D_POWER_MAX		35										// 最大２のべき乗数

// 内部変数
static FILE *szpFpI;											// 入力
static int si1FSize[D_FCNT];									// 必要サイズ[家族]
static int si1SCnt[D_FCNT + 1];									// シート数[家族]
static int si2Comb[D_COMB_MAX][D_COMB_MAX];						// 組み合わせ表[n][k] = nCk
static int si1Power2[D_POWER_MAX];								// ２のべき乗

// 内部変数 - テスト用
#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;
}

// nCk(組み合わせ表[n][k]) - 作成 - パスカルの三角形
int
fMakeComb(
	int piRng					// <I> 範囲
)
{
	int i, j;

	for (i = 0; i <= piRng; ++i) {
		for (j = 0; j <= i; ++j) {
			if (j == 0 || j == i) {
				si2Comb[i][j] = 1;
			}
			else {
				si2Comb[i][j] = si2Comb[i - 1][j - 1] + si2Comb[i - 1][j];
			}
		}
	}

	return 0;
}

// 実行メイン
int
fMain(
)
{
	int i, j;
	char lc1Buf[1024];

	// 必要サイズ - 取得
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d%d%d", &si1FSize[0], &si1FSize[1], &si1FSize[2]);

	// 必要サイズ - ソート
	qsort(si1FSize, D_FCNT, sizeof(int), fSortFnc);

	// シート数 - 取得
	int liSCnt;
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d", &liSCnt);
	for (i = 0; i < liSCnt; i++) {
		int liSize;
		fscanf(szpFpI, "%d", &liSize);
		for (j = 0; j < D_FCNT; j++) {
			if (liSize < si1FSize[j]) {
				break;
			}
		}
		si1SCnt[j]++;
	}
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);

	// nCk(組み合わせ表[n][k]) - 作成
	fMakeComb(liSCnt);

	// ２のべき乗 - 作成
	si1Power2[0] = 1;
	for (i = 1; i <= liSCnt; i++) {
		si1Power2[i] = si1Power2[i - 1] * 2;
	}

	// 全パターン数
	int liSum = 0;

	// ３家族OK１枚
	if (si1SCnt[3] >= 1) {

		// ３家族OK１枚・２家族OK２枚以上
		for (i = 2; i <= si1SCnt[2]; i++) {
			int liCnt = si1SCnt[3];
			liCnt *= si2Comb[si1SCnt[2]][i];
			liCnt *= si1Power2[si1SCnt[1] + si1SCnt[0]];
			liSum += liCnt;
		}

		// ３家族OK１枚・２家族OK１枚・１家族OK１枚以上
		if (si1SCnt[2] >= 1) {
			for (i = 1; i <= si1SCnt[1]; i++) {
				int liCnt = si1SCnt[3];
				liCnt *= si2Comb[si1SCnt[2]][1];
				liCnt *= si2Comb[si1SCnt[1]][i];
				liCnt *= si1Power2[si1SCnt[0]];
				liSum += liCnt;
			}
		}
	}

	// ３家族OK２枚
	if (si1SCnt[3] >= 2) {

		// ３家族OK２枚・２家族OK１枚以上
		for (i = 1; i <= si1SCnt[2]; i++) {
			int liCnt = si2Comb[si1SCnt[3]][2];
			liCnt *= si2Comb[si1SCnt[2]][i];
			liCnt *= si1Power2[si1SCnt[1] + si1SCnt[0]];
			liSum += liCnt;
		}

		// ３家族OK２枚・１家族OK１枚以上
		for (i = 1; i <= si1SCnt[1]; i++) {
			int liCnt = si2Comb[si1SCnt[3]][2];
			liCnt *= si2Comb[si1SCnt[1]][i];
			liCnt *= si1Power2[si1SCnt[0]];
			liSum += liCnt;
		}
	}

	// ３家族OK３枚以上
	for (i = 3; i <= si1SCnt[3]; i++) {
		int liCnt = si2Comb[si1SCnt[3]][i];
		liCnt *= si1Power2[si1SCnt[2] + si1SCnt[1] + si1SCnt[0]];
		liSum += liCnt;
	}

	return liSum;
}

// １回実行
int
fOne(
)
{
	int liRet;
	char lc1Buf[1024];

	// データ - 初期化
	memset(si1SCnt, 0, sizeof(si1SCnt));						// シート数

																// 入力 - セット
#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

	// 実行メイン
	liRet = fMain();

	// 結果 - セット
	sprintf(lc1Buf, "%d\n", liRet);

	// 結果 - 出力
	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