#305033 (Java) No.3 ビットすごろく

• 問題: No.3 ビットすごろく
• ユーザー: kusomushi
• 提出日時: 2018-12-18 19:59:49
• 言語: Java (openjdk 23)
• 結果: WA
• 実行時間: -
• コード長: 6,202 bytes
• コンパイル時間: 2,465 ms
• コンパイル使用メモリ: 80,824 KB
• 実行使用メモリ: 53,892 KB
• 最終ジャッジ日時: 2024-09-25 07:46:43
• 合計ジャッジ時間: 5,994 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

ファイルパターン	結果
other	AC * 32 WA * 1

ソースコード

import java.io.*;
import java.util.*;
import java.util.function.Function;

public class Main {

    static int N;

    public static void main(String[] args) {
        FastScanner sc = new FastScanner(System.in);
        N = sc.nextInt();

        System.out.println(solve());
    }

    static int solve() {
        int[] A = new int[N*2];
        int[] B = new int[N*2];
        for (int i = 0; i < N; i++) {
            int n = i+1;
            int cnt = Integer.bitCount(n);
            if( n + cnt <= N ) {
                A[i*2] = n;
                B[i*2] = n + cnt;
            }
            if( n - cnt >= 1 ) {
                A[i*2+1] = n;
                B[i*2+1] = n - cnt;
            }
        }
        int[][] G = adjD(N+1, A, B);

        ArrayDeque<Integer> q = new ArrayDeque<>();
        q.add(1);
        int[] dist = new int[N+1];
        Arrays.fill(dist, Integer.MAX_VALUE);
        dist[1] = 0;
        while( !q.isEmpty() ) {
            int a = q.poll();

            for (int next : G[a]) {
                if( dist[next] > dist[a] + 1 ) {
                    dist[next] = dist[a] + 1;
                    q.add(next);
                }
            }
        }
        if( dist[N] == Integer.MAX_VALUE ) {
            return -1;
        } else {
            return dist[N] + 1;
        }
    }

    static int[][] adjD(int n, int[] from, int[] to) {
        Set<Integer> set = new HashSet<>();

        int[][] adj = new int[n][];
        int[] cnt = new int[n];


        for (int i = 0; i < from.length; i++) {
            int f = from[i];
            int t = to[i];
            if( f == t ) continue;

            int key = Math.min(f, t) * 10000 + Math.max(f, t);
            if( set.contains(key) ) continue;

            cnt[f]++;
            set.add(key);
        }
        for (int i = 0; i < n; i++) {
            adj[i] = new int[cnt[i]];
        }

        set.clear();
        for (int i = 0; i < from.length; i++) {
            int f = from[i];
            int t = to[i];
            if( f == t ) continue;

            int key = Math.min(f, t) * 10000 + Math.max(f, t);
            if( set.contains(key) ) continue;

            adj[from[i]][--cnt[from[i]]] = to[i];
            set.add(key);
        }
        return adj;
    }

    @SuppressWarnings("unused")
    static class FastScanner {
        private BufferedReader reader;
        private StringTokenizer tokenizer;

        FastScanner(InputStream in) {
            reader = new BufferedReader(new InputStreamReader(in));
            tokenizer = null;
        }

        String next() {
            if (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }

        String nextLine() {
            if (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    return reader.readLine();
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken("\n");
        }

        long nextLong() {
            return Long.parseLong(next());
        }

        int nextInt() {
            return Integer.parseInt(next());
        }

        int[] nextIntArray(int n) {
            int[] a = new int[n];
            for (int i = 0; i < n; i++) a[i] = nextInt();
            return a;
        }

        int[] nextIntArray(int n, int delta) {
            int[] a = new int[n];
            for (int i = 0; i < n; i++) a[i] = nextInt() + delta;
            return a;
        }

        long[] nextLongArray(int n) {
            long[] a = new long[n];
            for (int i = 0; i < n; i++) a[i] = nextLong();
            return a;
        }
    }

    static <A> void writeLines(A[] as, Function<A, String> f) {
        PrintWriter pw = new PrintWriter(System.out);
        for (A a : as) pw.println(f.apply(a));
        pw.flush();
    }

    static void writeLines(int[] as) {
        PrintWriter pw = new PrintWriter(System.out);
        for (int a : as) pw.println(a);
        pw.flush();
    }

    static void writeLines(long[] as) {
        PrintWriter pw = new PrintWriter(System.out);
        for (long a : as) pw.println(a);
        pw.flush();
    }

    static void writeSingleLine(int[] as) {
        PrintWriter pw = new PrintWriter(System.out);
        for (int i = 0; i < as.length; i++) {
            if (i != 0) pw.print(" ");
            pw.print(i);
        }
        pw.println();
        pw.flush();
    }

    static int max(int... as) {
        int max = Integer.MIN_VALUE;
        for (int a : as) max = Math.max(a, max);
        return max;
    }

    static int min(int... as) {
        int min = Integer.MAX_VALUE;
        for (int a : as) min = Math.min(a, min);
        return min;
    }

    static int lowerBound(int[] array, int value) {
        int low = 0;
        int high = array.length;
        int mid;
        while (low < high) {
            mid = ((high - low) >>> 1) + low;
            if (array[mid] < value) low = mid + 1;
            else high = mid;
        }
        return low;
    }

    static int upperBound(long[] array, long value) {
        int low = 0;
        int high = array.length;
        int mid;
        while (low < high) {
            mid = ((high - low) >>> 1) + low;
            if (array[mid] <= value) low = mid + 1;
            else high = mid;
        }
        return low;
    }

    static void debug(Object... args) {
        StringJoiner j = new StringJoiner(" ");
        for (Object arg : args) {
            if (arg instanceof int[]) j.add(Arrays.toString((int[]) arg));
            else if (arg instanceof long[]) j.add(Arrays.toString((long[]) arg));
            else if (arg instanceof double[]) j.add(Arrays.toString((double[]) arg));
            else if (arg instanceof Object[]) j.add(Arrays.toString((Object[]) arg));
            else j.add(arg.toString());
        }
        System.err.println(j.toString());
    }
}