結果

問題 No.160 最短経路のうち辞書順最小
コンテスト
ユーザー bal4u
提出日時 2019-07-19 23:18:12
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 44 ms / 5,000 ms
コード長 2,637 bytes
コンパイル時間 1,207 ms
コンパイル使用メモリ 31,872 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-26 04:01:17
合計ジャッジ時間 1,185 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:8:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
    8 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:16:24: note: in expansion of macro 'gc'
   16 |         int n = 0, c = gc();
      |                        ^~
main.c: In function 'out':
main.c:9:15: warning: implicit declaration of function 'putchar_unlocked' [-Wimplicit-function-declaration]
    9 | #define pc(c) putchar_unlocked(c)
      |               ^~~~~~~~~~~~~~~~
main.c:25:17: note: in expansion of macro 'pc'
   25 |         if (!n) pc('0');
      |                 ^~

ソースコード

diff # 

// yukicoder: No.160 最短経路のうち辞書順最小
// 2019.7.19 bal4u

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

#if 1
#define gc() getchar_unlocked()
#define pc(c) putchar_unlocked(c)
#else
#define gc() getchar()
#define pc(c) putchar(c)
#endif

int in() {   // 非負整数の入力
	int n = 0, c = gc();
	do n = 10 * n + (c & 0xf); while ((c = gc()) >= '0');
	return n;
}

void out(int n) { // 非負整数の表示（出力）
	int i;
	char b[20];

	if (!n) pc('0');
	else {
		i = 0; while (n) b[i++] = n % 10 + '0', n /= 10;
		while (i--) pc(b[i]);
	}
}

//// 優先度付きキュー（ダイクストラ法用）
#define MAX 1000000
typedef struct { int t, n, pre; } 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 pre) {
	int i, min;
	QUE qt;

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

//// 本問題関連
#define INF 0x33333333
#define MAXV 202
int N, S, G;
int hi[MAXV], to[MAXV][MAXV], tm[MAXV][MAXV];
int vis[MAXV];
int prev[MAXV];
int a[MAXV], sz;

int path(int *a, int n, int pre) {
	int i, w = 0;
	a[w++] = n, n = pre;
	while (n >= 0) a[w++] = n, n = prev[n];
	return w;
}

void dijkstra() {
	int i, n, t, pre, mi;
	int tmp[MAXV], tsz;
	
	mi = INF, memset(vis, INF, sizeof(vis));
	enq(G, 0, -1);
	while (qsize) {
		n = que[0].n, t = que[0].t, pre = que[0].pre, deq();
		if (n == S) {
			if (t < mi) sz = path(a, n, pre), mi = t;
			else if (t > mi) return;
			tsz = path(tmp, n, pre);
			a[sz] = -1, tmp[tsz] = -1;
			for (i = 0; a[i] == tmp[i]; i++);
			if (tmp[i] < a[i]) {
				for (i = 0; i < tsz; i++) a[i] = tmp[i]; 
				sz = tsz;
			}
			continue;
		}
		if (vis[n] < t || vis[n] == t && prev[n] <= pre) continue;
		vis[n] = t, prev[n] = pre;
		for (i = 0; i < hi[n]; i++) {
			int nx = to[n][i];
			enq(nx, t+tm[n][i], n);
		}
	}
}

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

	N = in(), M = in(), S = in(), G = in();
	while (M--) {
		int a = in(), b = in(), c = in();
		k = hi[a]++, to[a][k] = b, tm[a][k] = c;
		k = hi[b]++, to[b][k] = a, tm[b][k] = c;
	}
	dijkstra();
	out(a[0]); for (i = 1; i < sz; i++) pc(' '), out(a[i]);	pc('\n');
	return 0;
}
0