結果

問題 No.2673 A present from B
ユーザー ks2mks2m
提出日時 2024-03-15 22:43:09
言語 Java
(openjdk 23)
結果
AC  
実行時間 217 ms / 2,000 ms
コード長 1,758 bytes
コンパイル時間 3,714 ms
コンパイル使用メモリ 80,068 KB
実行使用メモリ 57,900 KB
最終ジャッジ日時 2024-09-30 02:10:23
合計ジャッジ時間 7,293 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Deque;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) throws Exception {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int m = sc.nextInt();
		int[] a = new int[m];
		for (int i = 0; i < m; i++) {
			a[m - 1 - i] = sc.nextInt() - 1;
		}
		sc.close();

		int inf = 100000000;
		int[][] d = new int[m + 1][n];
		for (int j = 0; j < d.length; j++) {
			Arrays.fill(d[j], inf);
		}
		d[0][0] = 0;
		Deque<Obj> que = new ArrayDeque<>();
		que.add(new Obj(0, 0, 0));
		while (!que.isEmpty()) {
			Obj o = que.poll();
			if (o.d != d[o.x][o.y]) {
				continue;
			}
			if (o.x < m) {
				if (a[o.x] == o.y) {
					if (o.d < d[o.x + 1][o.y + 1]) {
						d[o.x + 1][o.y + 1] = o.d;
						que.addFirst(new Obj(o.x + 1, o.y + 1, o.d));
					}
				} else if (a[o.x] == o.y - 1) {
					if (o.d < d[o.x + 1][o.y - 1]) {
						d[o.x + 1][o.y - 1] = o.d;
						que.addFirst(new Obj(o.x + 1, o.y - 1, o.d));
					}
				} else {
					if (o.d < d[o.x + 1][o.y]) {
						d[o.x + 1][o.y] = o.d;
						que.addFirst(new Obj(o.x + 1, o.y, o.d));
					}
				}
			}
			if (o.y < n - 1 && o.d + 1 < d[o.x][o.y + 1]) {
				d[o.x][o.y + 1] = o.d + 1;
				que.addLast(new Obj(o.x, o.y + 1, o.d + 1));
			}
			if (o.y > 0 && o.d + 1 < d[o.x][o.y - 1]) {
				d[o.x][o.y - 1] = o.d + 1;
				que.addLast(new Obj(o.x, o.y - 1, o.d + 1));
			}
		}
		StringBuilder sb = new StringBuilder();
		for (int i = 1; i < n; i++) {
			sb.append(d[m][i]).append(' ');
		}
		sb.deleteCharAt(sb.length() - 1);
		System.out.println(sb.toString());
	}

	static class Obj {
		int x, y, d;

		public Obj(int x, int y, int d) {
			this.x = x;
			this.y = y;
			this.d = d;
		}
	}
}
0