結果

問題 No.160 最短経路のうち辞書順最小
コンテスト
ユーザー asugen0402
提出日時 2019-01-30 16:16:06
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 286 ms / 5,000 ms
コード長 6,211 bytes
コンパイル時間 1,001 ms
コンパイル使用メモリ 32,512 KB
実行使用メモリ 10,112 KB
最終ジャッジ日時 2024-10-15 01:30:00
合計ジャッジ時間 2,539 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

// 内部定数
#define D_STA_MAX		200										// 最大駅数
#define D_HEAP_MAX		D_STA_MAX * D_STA_MAX					// 最大ヒープ数

// 内部構造体 - ヒープ情報
typedef struct Heap {
	int miTime;													// 移動時間
	int mi1SNo[D_STA_MAX];										// 通過駅
	int miSCnt;													// 通過駅数
} Heap;

// 内部構造体 - 駅情報
typedef struct Sta {
	int mi1Time[D_STA_MAX];										// 移動時間[駅]
	Heap mzMin;													// 最短データ
} Sta;

// 内部変数
static FILE *szpFpI;											// 入力
static Sta sz1Sta[D_STA_MAX];									// 駅
static int siSCnt;												// 駅数
static Heap sz1Heap[D_HEAP_MAX];								// ヒープ
static int siHCnt;												// ヒープ数

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

// ヒープ - 比較 - 移動時間昇順 - 通過駅昇順
int
fHeapCmp(
	Heap *pzpHeap1				// <I> データ１
	, Heap *pzpHeap2			// <I> データ２
)
{
	// 移動時間昇順
	if (pzpHeap1->miTime < pzpHeap2->miTime) {
		return -1;
	}
	else if (pzpHeap1->miTime > pzpHeap2->miTime) {
		return 1;
	}

	// 通過駅昇順
	for (int liNo = 0; ; liNo++) {
		if (pzpHeap1->miSCnt <= liNo) {
			return -1;
		}
		else if (pzpHeap2->miSCnt <= liNo) {
			return 1;
		}
		if (pzpHeap1->mi1SNo[liNo] < pzpHeap2->mi1SNo[liNo]) {
			return -1;
		}
		else if (pzpHeap1->mi1SNo[liNo] > pzpHeap2->mi1SNo[liNo]) {
			return 1;
		}
	}

	return 0;
}

// ヒープ - 親子関係チェック
// 戻り値：[>=0]:変更した子の配列番号 [-1]:変更なし
int
fHeapChk(
	int piPNo					// <I> 親の配列番号 0～
)
{
	int liRet;

	// 最小値
	int liMNo = piPNo;

	// 左の子と比較
	int liCNo = piPNo * 2 + 1;
	if (liCNo < siHCnt) {
		liRet = fHeapCmp(&sz1Heap[liMNo], &sz1Heap[liCNo]);
		if (liRet == 1) {
			liMNo = liCNo;
		}
	}

	// 右の子と比較
	liCNo = piPNo * 2 + 2;
	if (liCNo < siHCnt) {
		liRet = fHeapCmp(&sz1Heap[liMNo], &sz1Heap[liCNo]);
		if (liRet == 1) {
			liMNo = liCNo;
		}
	}

	// 自分が最小値であるかチェック
	if (piPNo == liMNo) {
		return -1;
	}

	// 値の交換
	Heap lzWork;
	memcpy(&lzWork, &sz1Heap[liMNo], sizeof(Heap));
	memcpy(&sz1Heap[liMNo], &sz1Heap[piPNo], sizeof(Heap));
	memcpy(&sz1Heap[piPNo], &lzWork, sizeof(Heap));

	return liMNo;
}

// ヒープ - キュー追加
int
fHeapEnqueue(
	Heap *pzpHeap				// <I> データ
)
{
	int liRet;

	// 現在駅
	int liSNo = pzpHeap->mi1SNo[pzpHeap->miSCnt - 1];

	// 最短データ
	liRet = fHeapCmp(pzpHeap, &sz1Sta[liSNo].mzMin);
	if (liRet >= 0) {
		return 0;
	}
	memcpy(&sz1Sta[liSNo].mzMin, pzpHeap, sizeof(Heap));

	// 末尾に追加
	memcpy(&sz1Heap[siHCnt], pzpHeap, sizeof(Heap));
	siHCnt++;

	// 親子関係チェック
	int liNo = siHCnt - 1;
	while (1) {

		// 親の配列番号
		liNo = (liNo - 1) / 2;

		// 親子関係チェック
		liRet = fHeapChk(liNo);
		if (liRet < 0) {
			break;
		}
	}

	return 0;
}

// ヒープ - キュー取得
int
fHeapDequeue(
	Heap *pzpRet				// <O> 取得先
)
{
	// データ数
	if (siHCnt < 1) {
		return -1;
	}

	// 取得
	memcpy(pzpRet, &sz1Heap[0], sizeof(Heap));
	siHCnt--;

	// データ数
	if (siHCnt < 1) {
		return 0;
	}

	// 末尾を先頭へ
	memcpy(&sz1Heap[0], &sz1Heap[siHCnt], sizeof(Heap));

	// 親子関係チェック
	int liNo = 0;
	while (liNo >= 0) {
		liNo = fHeapChk(liNo);
	}

	return 0;
}

// 実行メイン
int
fMain(
	int piTNo					// <I> テスト番号 1～
)
{
	int i, liRet;
	char lc1Buf[1024], lc1Out[1024];

	// データ - 初期化
	memset(sz1Sta, 0, sizeof(sz1Sta));							// 駅

	// 入力 - セット
#ifdef D_TEST
	sprintf(lc1Buf, ".\\Test\\T%d.txt", piTNo);
	szpFpI = fopen(lc1Buf, "r");
	sprintf(lc1Buf, ".\\Test\\A%d.txt", piTNo);
	szpFpA = fopen(lc1Buf, "r");
	siRes = 0;
#else
	szpFpI = stdin;
#endif

	// 駅数・線路数・出発駅・到着駅 - 取得
	int liRCnt, liStart, liGoal;
	fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
	sscanf(lc1Buf, "%d%d%d%d", &siSCnt, &liRCnt, &liStart, &liGoal);

	// 最短時間 - 初期化
	for (i = 0; i < siSCnt; i++) {
		sz1Sta[i].mzMin.miTime = INT_MAX;
	}

	// 線路 - 取得
	for (i = 0; i < liRCnt; i++) {
		int liSNo1, liSNo2, liTime;
		fgets(lc1Buf, sizeof(lc1Buf), szpFpI);
		sscanf(lc1Buf, "%d%d%d", &liSNo1, &liSNo2, &liTime);

		// 移動時間 - セット
		sz1Sta[liSNo1].mi1Time[liSNo2] = liTime;
		sz1Sta[liSNo2].mi1Time[liSNo1] = liTime;
	}

	// 出発駅 - 登録
	Heap lzHeap;
	memset(&lzHeap, 0, sizeof(lzHeap));
	lzHeap.mi1SNo[0] = liStart;
	lzHeap.miSCnt = 1;
	fHeapEnqueue(&lzHeap);

	// 算出
	while (1) {

		// キュー取得
		liRet = fHeapDequeue(&lzHeap);
		if (liRet != 0) {
			break;
		}

		// 現在駅
		int liSNo = lzHeap.mi1SNo[lzHeap.miSCnt - 1];

		// 隣接駅へ
		for (i = 0; i < siSCnt; i++) {
			int liTime = sz1Sta[liSNo].mi1Time[i];
			if (liTime < 1) {
				continue;
			}

			// データコピー
			Heap lzHeap2;
			memcpy(&lzHeap2, &lzHeap, sizeof(lzHeap2));

			// 隣接駅 - 追加
			lzHeap2.mi1SNo[lzHeap2.miSCnt] = i;
			lzHeap2.miSCnt++;
			lzHeap2.miTime += liTime;

			// 登録
			fHeapEnqueue(&lzHeap2);
		}
	}

	// 結果 - セット
	memcpy(&lzHeap, &sz1Sta[liGoal].mzMin, sizeof(lzHeap));
	sprintf(lc1Out, "%d", lzHeap.mi1SNo[0]);
	for (i = 1; i < lzHeap.miSCnt; i++) {
		sprintf(lc1Buf, " %d", lzHeap.mi1SNo[i]);
		strcat(lc1Out, lc1Buf);
	}
	strcat(lc1Out, "\n");

	// 結果 - 表示
#ifdef D_TEST
	fgets(lc1Buf, sizeof(lc1Buf), szpFpA);
	if (strcmp(lc1Buf, lc1Out)) {
		siRes = -1;
	}
#else
	printf("%s", lc1Out);
#endif

	// 残データ有無
#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", piTNo);
	}
	else {
		printf("NG %d\n", piTNo);
	}
#endif

	return 0;
}

int
main()
{

#ifdef D_TEST
	int i;
	for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) {
		fMain(i);
	}
#else
	fMain(0);
#endif

	return 0;
}
0