結果

問題 No.1492 01文字列と転倒
ユーザー tenten
提出日時 2021-05-06 13:19:23
言語 Java
(openjdk 23)
結果
AC  
実行時間 3,960 ms / 4,000 ms
コード長 2,346 bytes
コンパイル時間 2,572 ms
コンパイル使用メモリ 80,560 KB
実行使用メモリ 97,400 KB
最終ジャッジ日時 2024-06-23 22:00:28
合計ジャッジ時間 39,228 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.*;
import java.util.*;

public class Main {
    static ArrayList<ArrayList<Integer>> lists = new ArrayList<>();
    public static void main(String[] args) throws Exception {
        Scanner sc = new Scanner();
        int n = sc.nextInt();
        int m = sc.nextInt();
        HashMap<Integer, HashMap<Integer, Integer>> current = new HashMap<>();
        current.put(0, new HashMap<>());
        current.get(0).put(0, 1);
        for (int i = 2; i <= n * 2; i++) {
            HashMap<Integer, HashMap<Integer, Integer>> next = new HashMap<>();
            for (int x : current.keySet()) {
                for (int y : current.get(x).keySet()) {
                    if (i - x <= n) {
                        if (!next.containsKey(x)) {
                            next.put(x, new HashMap<>());
                        }
                        next.get(x).put(y + x, (next.get(x).getOrDefault(y + x, 0) + current.get(x).get(y)) % m);
                    }
                    if (x < i - x - 1) {
                        if (!next.containsKey(x + 1)) {
                            next.put(x + 1, new HashMap<>());
                        }
                        next.get(x + 1).put(y, (next.get(x + 1).getOrDefault(y, 0) + current.get(x).get(y)) % m);
                    }
                }
            }
            current = next;
        }
        int[] ans = new int[n * n + 1];
        for (HashMap<Integer, Integer> map : current.values()) {
            for (int x : map.keySet()) {
                ans[x] += map.get(x);
                ans[x] %= m;
            }
        }
        StringBuilder sb = new StringBuilder();
        for (int x : ans) {
            sb.append(x).append("\n");
        }
        System.out.print(sb);
    }
}

class Scanner {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    StringTokenizer st = new StringTokenizer("");
    
    public Scanner() throws Exception {
        
    }
    
    public int nextInt() throws Exception {
        return Integer.parseInt(next());
    }
    
    public long nextLong() throws Exception {
        return Long.parseLong(next());
    }
    
    public String next() throws Exception {
        if (!st.hasMoreTokens()) {
            st = new StringTokenizer(br.readLine());
        }
        return st.nextToken();
    }
}
0