#351148 (C) No.368 LCM of K-products

提出ソース
結果

問題	No.368 LCM of K-products
コンテスト
ユーザー	asugen0402
提出日時	2019-06-12 15:26:16
言語	C (gcc 13.3.0)
結果	WA
実行時間	-
コード長	8,451 bytes
コンパイル時間	453 ms
コンパイル使用メモリ	34,432 KB
実行使用メモリ	201,216 KB
最終ジャッジ日時	2024-10-10 20:35:32
合計ジャッジ時間	9,571 ms
ジャッジサーバーID （参考情報）	judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン	結果
other	AC * 19 WA * 16
権限があれば一括ダウンロードができます
ソースコード

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

// 内部定数
#define D_ON			1										// 汎用フラグ - ON
#define D_OFF			0										// 汎用フラグ - OFF
#define D_MOD			1000000007								// 除数(10の9乗+7)
#define D_ARRAY_MAX		1000									// 最大配列数
#define D_PRM_MAX		10000									// 最大素数リスト数
#define D_PRM_WORK		100000									// 素数リスト作成用ワーク
#define D_SEGT_CNT		2048									// セグメントツリーデータ数(2の累乗)
#define D_SEGT_KIND		D_PRM_MAX								// セグメントツリーデータ種類数

// 内部構造体 - 値情報
typedef struct Val {
	int miNo;													// 配列番号 0～
	int miVal;													// 値
	int mi1PCnt[D_PRM_MAX];										// 素数数[素数リスト番号]
} Val;

// 内部構造体 - セグメントツリー情報
typedef struct SegT {
	int mi1Val[D_SEGT_KIND];									// 値
} SegT;

// 内部変数
static FILE *szpFpI;											// 入力
static Val sz1Array[D_ARRAY_MAX];								// 配列
static int siACnt;												// 配列数
static int siMCnt;												// 積数
static int si1Prm[D_PRM_MAX];									// 素数リスト
static char sc1PWork[D_PRM_WORK];								// 素数リスト作成用ワーク
static int siPCnt;												// 素数リスト数
static SegT sz1SegT[D_SEGT_CNT * 2];							// セグメントツリー
static int siSCNo;												// セグメントツリー - 子の開始位置

// 内部変数 - テスト用
#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
fGetMax(
	int piVal1					// <I> 値１
	, int piVal2				// <I> 値２
)
{
	if (piVal1 > piVal2) {
		return piVal1;
	}
	else {
		return piVal2;
	}
}

// ソート関数 - 値昇順
int
fSortFncVu(
	const void *pzpVal1			// <I> 値１
	, const void *pzpVal2		// <I> 値２
)
{
	Val *lzpVal1 = (Val *)pzpVal1;
	Val *lzpVal2 = (Val *)pzpVal2;

	// 値昇順
	if (lzpVal1->miVal > lzpVal2->miVal) {
		return 1;
	}
	else if (lzpVal1->miVal < lzpVal2->miVal) {
		return -1;
	}

	return 0;
}

// ソート関数 - No昇順
int
fSortFncNu(
	const void *pzpVal1			// <I> 値１
	, const void *pzpVal2		// <I> 値２
)
{
	Val *lzpVal1 = (Val *)pzpVal1;
	Val *lzpVal2 = (Val *)pzpVal2;

	// No昇順
	if (lzpVal1->miNo > lzpVal2->miNo) {
		return 1;
	}
	else if (lzpVal1->miNo < lzpVal2->miNo) {
		return -1;
	}

	return 0;
}

// セグメントツリー - 子の開始位置 - セット
int
fSegTSetCSNo(
	int piCnt					// <I> 子データ数
)
{
	siSCNo = 1;
	while (siSCNo < piCnt) {
		siSCNo *= 2;
	}

	return 0;
}

// セグメントツリー - ２つの値の処理
int
fSegTVal2(
	int piVal1					// <I> 値１
	, int piVal2				// <I> 値２
)
{
	return piVal1 + piVal2;
}

// セグメントツリー - 親データセット
int
fSegTSetPData(
	int piKind					// <I> 種類
)
{
	int i;

	// 最初の親番号範囲
	int liPNo1 = siSCNo / 2;
	int liPNo2 = siSCNo - 1;

	// 作成
	while (liPNo1 > 0) {
		for (i = liPNo1; i <= liPNo2; i++) {
			sz1SegT[i].mi1Val[piKind] = fSegTVal2(sz1SegT[i * 2].mi1Val[piKind], sz1SegT[i * 2 + 1].mi1Val[piKind]);
		}

		// 次の親番号範囲
		liPNo2 = liPNo1 - 1;
		liPNo1 /= 2;
	}

	return 0;
}

// セグメントツリー - 範囲取得
int
fSegTGetRngMain(
	int piKind					// <I> 種類
	, int piGetS				// <I> 取得範囲 - 開始 0～
	, int piGetE				// <I> 取得範囲 - 終了 0～
	, int piNNo					// <I> 現在位置 1～
	, int piNowS				// <I> 現在範囲 - 開始 0～
	, int piNowE				// <I> 現在範囲 - 終了 0～
)
{
	// 内包チェック
	if (piGetS <= piNowS && piNowE <= piGetE) {
		return sz1SegT[piNNo].mi1Val[piKind];
	}

	// 中間位置
	int liCenter = (piNowS + piNowE) / 2;

	// 初期値
	int liVal = 0;

	// 左側
	if (piGetS <= liCenter) {
		liVal = fSegTGetRngMain(piKind, piGetS, piGetE, piNNo * 2, piNowS, liCenter);
	}

	// 右側
	if (piGetE >= liCenter + 1) {
		int liVal2 = fSegTGetRngMain(piKind, piGetS, piGetE, piNNo * 2 + 1, liCenter + 1, piNowE);
		liVal = fSegTVal2(liVal, liVal2);
	}

	return liVal;
}
int
fSegTGetRng(
	int piKind					// <I> 種類
	, int piGetS				// <I> 取得範囲 - 開始 0～
	, int piGetE				// <I> 取得範囲 - 終了 0～
)
{
	return fSegTGetRngMain(piKind, piGetS, piGetE, 1, 0, siSCNo - 1);
}

// 配列の素因数分解
int
fDivPrm(
	int piMax					// <I> 最大値
)
{
	int i;

	// 初期化
	siPCnt = 0;
	memset(si1Prm, 0, sizeof(si1Prm));
	memset(sc1PWork, D_OFF, sizeof(sc1PWork));

	// 開始値
	int liNow = 2;

	// 上限値
	int liLimit = (int)sqrt((double)piMax);

	// 作成 - 開始
	while (1) {

		// 素数の検索
		while (sc1PWork[liNow] != D_OFF) {
			liNow++;
		}

		// 上限チェック
		if (liNow > liLimit) {
			break;
		}

		// 配列の素因数分解
		for (i = 0; i < siACnt; i++) {
			while (sz1Array[i].miVal % liNow == 0) {
				sz1Array[i].miVal /= liNow;
				sz1Array[i].mi1PCnt[siPCnt]++;
				si1Prm[siPCnt] = liNow;
			}
		}

		// 素数リスト - 追加
		if (si1Prm[siPCnt] > 0) {
			siPCnt++;
		}

		// ワークに素数以外をセット
		int liVal = liNow;
		while (liVal <= liLimit) {
			sc1PWork[liVal] = D_ON;
			liVal += liNow;
		}
	}

	// 配列 - ソート
	qsort(sz1Array, siACnt, sizeof(Val), fSortFncVu);

	// 残った素数を追加
	for (i = 0; i < siACnt; i++) {
		if (sz1Array[i].miVal > 1) {
			if (si1Prm[siPCnt - 1] != sz1Array[i].miVal) {
				si1Prm[siPCnt] = sz1Array[i].miVal;
				siPCnt++;
			}
			sz1Array[i].mi1PCnt[siPCnt - 1]++;
		}
	}

	// 配列 - ソート
	qsort(sz1Array, siACnt, sizeof(Val), fSortFncNu);

	return 0;
}

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

	// 配列数・積数 - 取得
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d%d", &siACnt, &siMCnt);

	// 配列 - 取得
	int liMax = 0;
	for (i = 0; i < siACnt; i++) {
		sz1Array[i].miNo = i;
		fscanf(szpFpI, "%d", &sz1Array[i].miVal);
		liMax = fGetMax(liMax, sz1Array[i].miVal);
	}
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);

	// 配列の素因数分解
	fDivPrm(liMax);

	// セグメントツリー - 子の開始位置 - セット
	fSegTSetCSNo(siACnt * 2);

	// セグメントツリー - 子データセット
	for (i = 0; i < 2; i++) {
		for (j = 0; j < siACnt; j++) {
			for (k = 0; k < siPCnt; k++) {
				sz1SegT[siSCNo + siACnt * i + j].mi1Val[k] = sz1Array[j].mi1PCnt[k];
			}
		}
	}

	// セグメントツリー - 親データセット
	for (i = 0; i < siPCnt; i++) {
		fSegTSetPData(i);
	}

	// １つ目 - 取得
	int li1Cnt[D_PRM_MAX];
	for (i = 0; i < siPCnt; i++) {
		li1Cnt[i] = fSegTGetRng(i, 0, siMCnt - 1);
	}

	// ２つ目以降 - 個数更新
	for (i = 1; i < siACnt; i++) {
		for (j = 0; j < siPCnt; j++) {
			int liVal = fSegTGetRng(j, i, i + siMCnt - 1);
			li1Cnt[j] = fGetMax(li1Cnt[j], liVal);
		}
	}

	// 最小公倍数 - 取得
	long long llLcm = 1;
	for (i = 0; i < siPCnt; i++) {
		for (j = 0; j < li1Cnt[i]; j++) {
			llLcm *= si1Prm[i];
			llLcm %= D_MOD;
		}
	}

	return (int)llLcm;
}

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

	// データ - 初期化
	memset(sz1Array, 0, sizeof(sz1Array));						// 配列
	memset(sz1SegT, 0, sizeof(sz1SegT));						// セグメントツリー

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

結果

ソースコード
