結果

問題 No.430 文字列検索
ユーザー asugen0402
提出日時 2019-06-13 10:36:14
言語 C
(gcc 8.3.0)
結果
AC  
実行時間 16 ms
コード長 5,409 Byte
コンパイル時間 277 ms
使用メモリ 8,916 KB
最終ジャッジ日時 2019-10-07 13:54:44

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
challenge01.txt AC 2 ms
6,872 KB
challenge02.txt AC 16 ms
6,872 KB
challenge03.txt AC 12 ms
6,876 KB
challenge04.txt AC 8 ms
6,876 KB
sample1.txt AC 2 ms
6,872 KB
sample2.txt AC 2 ms
6,872 KB
sample3.txt AC 2 ms
6,872 KB
sample4.txt AC 2 ms
6,872 KB
test1.txt AC 5 ms
6,876 KB
test2.txt AC 2 ms
6,876 KB
test3.txt AC 2 ms
6,876 KB
test4.txt AC 12 ms
6,876 KB
test5.txt AC 12 ms
8,916 KB
test6.txt AC 12 ms
6,872 KB
test7.txt AC 9 ms
6,872 KB
test8.txt AC 8 ms
6,872 KB
test9.txt AC 6 ms
6,876 KB
test10.txt AC 6 ms
6,872 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_CHAR_MAX		50005									// 最大文字数
#define D_VTX_MAX		50000									// 最大頂点数
#define D_EDGE_MAX		D_VTX_MAX								// 最大辺数
#define D_STACK_MAX		D_VTX_MAX								// 最大スタック数

// 内部構造体 - 辺情報
typedef struct Edge {
	int miVNo;													// 接続先頂点
	struct Edge *mzpNext;										// 次の辺情報
} Edge;

// 内部構造体 - 頂点情報
typedef struct Vtx {
	char mcChar;												// 頂点文字
	Edge *mzpEdge;												// 辺
	int miECnt;													// 最終文字数
} Vtx;

// 内部変数
static FILE *szpFpI;											// 入力
static char sc1Str[D_CHAR_MAX];									// 文字列S
static Vtx sz1Vtx[D_VTX_MAX];									// 頂点
static int siVCnt;												// 頂点数
static Edge sz1Edge[D_EDGE_MAX * 2];							// 辺
static int siECnt;												// 辺数
static int si1Stack[D_STACK_MAX];								// スタック
static int siSCnt;												// スタック数
static int siSSNo;												// スタック位置 - セット
static int siSGNo;												// スタック位置 - 取得

// 内部変数 - テスト用
#ifdef D_TEST
	static int siRes;
	static FILE *szpFpA;
	static int siTNo;
#endif

// 出力
int
fOut(
	char *pcpLine				// <I> 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
fAddEdge(
	int piVFNo					// <I> 頂点 - 元 0~
	, int piVTNo				// <I> 頂点 - 先 0~
)
{
	sz1Edge[siECnt].miVNo = piVTNo;
	sz1Edge[siECnt].mzpNext = sz1Vtx[piVFNo].mzpEdge;
	sz1Vtx[piVFNo].mzpEdge = &sz1Edge[siECnt];
	siECnt++;

	return 0;
}

// 頂点 - 検索
int
fSrhVtx(
	int piVNo					// <I> 頂点 0~
	, char pcChar				// <I> 文字
)
{
	// 辺でループ
	Edge *lzpEdge = sz1Vtx[piVNo].mzpEdge;
	while (lzpEdge != NULL) {
		if (sz1Vtx[lzpEdge->miVNo].mcChar == pcChar) {
			return lzpEdge->miVNo;
		}

		// 次の辺へ
		lzpEdge = lzpEdge->mzpNext;
	}

	return -1;
}

// 文字列 - 追加
int
fAddStr(
	int piVNo					// <I> 頂点 0~
	, char *pcpStr				// <I> 文字列
)
{
	// 文字チェック
	if (pcpStr[0] < 'A' || 'Z' < pcpStr[0]) {
		sz1Vtx[piVNo].miECnt++;
		return 0;
	}

	// 頂点 - 検索
	int liVNo = fSrhVtx(piVNo, pcpStr[0]);
	if (liVNo < 0) {
		liVNo = siVCnt;
		siVCnt++;
		sz1Vtx[liVNo].mcChar = pcpStr[0];
		fAddEdge(piVNo, liVNo);
	}

	// 次の文字へ
	fAddStr(liVNo, &pcpStr[1]);

	return 0;
}

// スタック - セット
int
fStackSet(
	int piVal					// <I> 値
)
{
	// セット
	si1Stack[siSSNo] = piVal;

	// スタック数
	siSCnt++;

	// スタック位置 - セット
	if (siSSNo < D_STACK_MAX - 1) {
		siSSNo++;
	}
	else {
		siSSNo = 0;
	}

	return 0;
}

// スタック - 取得
int
fStackGet(
)
{
	// スタック数
	if (siSCnt < 1) {
		return -1;
	}
	siSCnt--;

	// 取得値
	int liRet = si1Stack[siSGNo];

	// スタック位置 - 取得
	if (siSGNo < D_STACK_MAX - 1) {
		siSGNo++;
	}
	else {
		siSGNo = 0;
	}

	return liRet;
}

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

	// 文字列S - 取得
	fgets(sc1Str, sizeof(sc1Str), szpFpI);

	// 文字列C数
	int liSCnt;
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d", &liSCnt);

	// 文字列C - 取得
	for (i = 0; i < liSCnt; i++) {
		fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
		fAddStr(0, lc1Buf);
	}

	// スタック - 開始頂点
	fStackSet(0);

	// 文字列S - ループ
	int liMCnt = 0;
	for (i = 0; ; i++) {
		if (sc1Str[i] < 'A' || 'Z' < sc1Str[i]) {
			break;
		}

		// スタック数
		liSCnt = siSCnt;

		// スタック - 開始頂点
		fStackSet(0);

		// スタック - ループ
		for (j = 0; j < liSCnt; j++) {
			int liVNo = fStackGet();

			// 頂点 - 検索
			int liVNo2 = fSrhVtx(liVNo, sc1Str[i]);
			if (liVNo2 < 0) {
				continue;
			}

			// 最終文字数
			liMCnt += sz1Vtx[liVNo2].miECnt;

			// スタック - 追加
			fStackSet(liVNo2);
		}
	}

	return liMCnt;
}

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

	// データ - 初期化
	memset(sz1Vtx, 0, sizeof(sz1Vtx));							// 頂点
	siVCnt = 1;													// 頂点数
	siECnt = 0;													// 辺数
	siSCnt = 0;													// スタック数
	siSSNo = 0;													// スタック位置 - セット
	siSGNo = 0;													// スタック位置 - 取得

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