結果

問題 No.1 道のショートカット
コンテスト
ユーザー bal4u
提出日時 2019-04-07 16:03:47
言語 C
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,452 bytes
コンパイル時間 247 ms
コンパイル使用メモリ 31,616 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-08 05:10:10
合計ジャッジ時間 4,874 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 9 WA * 3 RE * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:11:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
   11 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:17:24: note: in expansion of macro 'gc'
   17 |         int n = 0, c = gc();
      |                        ^~

ソースコード

diff # 

// yukicoder: No.1 道のショートカット
// 2019.4.7 bal4u
// 最短距離に関する問題。ダイクストラ法でやってみる

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

//// 高速入力
#if 1
#define gc() getchar_unlocked()
#else
#define gc() getchar()
#endif
int in()    // 非負整数の入力
{
	int n = 0, c = gc();
	//	while (isspace(c)) c = gc();
	do n = 10 * n + (c & 0xf), c = gc(); while (c >= '0');
	return n;
}


//// 優先度付きキュー（ダイクストラ法用）
#define MAX 1000000
typedef struct { int t, n, money; } QUE;
QUE que[MAX]; int qsize;

#define PARENT(i) ((i)>>1)
#define LEFT(i)   ((i)<<1)
#define RIGHT(i)  (((i)<<1)+1)

void min_heapify(int i)
{
	int l, r, min;
	QUE qt;

	l = LEFT(i), r = RIGHT(i);
	if (l < qsize && que[l].t < que[i].t) min = l; else min = i;
	if (r < qsize && que[r].t < que[min].t) min = r;
	if (min != i) {
		qt = que[i]; que[i] = que[min]; que[min] = qt;
		min_heapify(min);
	}
}

void deq()
{
	que[0] = que[--qsize];
	min_heapify(0);
}

void enq(int n, int t, int money)
{
	int i, min;
	QUE qt;

	i = qsize++;
	que[i].t = t, que[i].n = n, que[i].money = money;
	while (i > 0 && que[min = PARENT(i)].t > que[i].t) {
		qt = que[i]; que[i] = que[min]; que[min] = qt;
		i = min;
	}
}


// 本問題関連
#define MAXN  55
int C; // 使える金額の上限
int N; // 町数
int V; // 道路の本数
int hi[MAXN], *to[MAXN], *money[MAXN], *tim[MAXN];
char vis[MAXN];
int S[MAXN], T[MAXN], Y[MAXN], M[MAXN];

int dijkstra(int start, int goal)
{
	int i, n, t, m, nx;

	enq(start, 0, 0);
	while (qsize) {
		n = que[0].n, t = que[0].t, m = que[0].money, deq();
		if (n == goal) return t;
		if (vis[n]) continue;
		vis[n] = 1;

		for (i = 0; i < hi[n]; i++) {
			if (m + money[n][i] > C) continue;
			nx = to[n][i];
			if (!vis[nx]) enq(nx, t + tim[n][i], m + money[n][i]);
		}
	}
	return -1;
}

int main()
{
	int i, k, s;

	N = in(), C = in(), V = in();
	for (i = 0; i < V; i++) S[i] = in(), hi[S[i]]++;
	for (i = 0; i < V; i++) T[i] = in();
	for (i = 0; i < V; i++) Y[i] = in();
	for (i = 0; i < V; i++) M[i] = in();
	for (i = 1; i <= N; i++) if (hi[i]) {
		to[i] = malloc(sizeof(int)*hi[i]);
		money[i] = malloc(sizeof(int)*hi[i]);
		tim[i] = malloc(sizeof(int)*hi[i]);
	}
	memset(hi, 0, sizeof(hi));
	for (i = 0; i < V; i++) {
		s = S[i];
		k = hi[s]++;
		to[s][k] = T[i], money[s][k] = Y[i], tim[s][k] = M[i];
	}

	printf("%d\n", dijkstra(1, N));
	return 0;
}
0