結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
ユーザー mikitmikit
提出日時 2020-05-29 22:22:54
言語 Java21
(openjdk 21)
結果
AC  
実行時間 417 ms / 2,000 ms
コード長 4,428 bytes
コンパイル時間 2,682 ms
コンパイル使用メモリ 78,100 KB
実行使用メモリ 57,592 KB
最終ジャッジ日時 2024-04-23 23:15:57
合計ジャッジ時間 7,663 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
50,020 KB
testcase_01 AC 48 ms
50,248 KB
testcase_02 AC 49 ms
50,156 KB
testcase_03 AC 50 ms
50,312 KB
testcase_04 AC 48 ms
50,124 KB
testcase_05 AC 51 ms
50,048 KB
testcase_06 AC 49 ms
50,368 KB
testcase_07 AC 52 ms
50,140 KB
testcase_08 AC 317 ms
54,796 KB
testcase_09 AC 63 ms
49,884 KB
testcase_10 AC 406 ms
57,568 KB
testcase_11 AC 197 ms
54,592 KB
testcase_12 AC 156 ms
52,228 KB
testcase_13 AC 204 ms
54,388 KB
testcase_14 AC 209 ms
54,628 KB
testcase_15 AC 317 ms
56,896 KB
testcase_16 AC 139 ms
52,472 KB
testcase_17 AC 146 ms
53,112 KB
testcase_18 AC 295 ms
54,972 KB
testcase_19 AC 209 ms
54,412 KB
testcase_20 AC 54 ms
50,392 KB
testcase_21 AC 261 ms
54,748 KB
testcase_22 AC 54 ms
50,068 KB
testcase_23 AC 50 ms
50,268 KB
testcase_24 AC 56 ms
49,956 KB
testcase_25 AC 417 ms
57,592 KB
testcase_26 AC 389 ms
57,228 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.OutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.BufferedOutputStream;
import java.io.UncheckedIOException;
import java.nio.charset.Charset;
import java.util.StringTokenizer;
import java.io.Writer;
import java.io.OutputStreamWriter;
import java.io.BufferedReader;
import java.io.InputStream;

/**
 * Built using CHelper plug-in
 * Actual solution is at the top
 *
 * @author mikit
 */
public class Main {
    public static void main(String[] args) {
        InputStream inputStream = System.in;
        OutputStream outputStream = System.out;
        LightScanner in = new LightScanner(inputStream);
        LightWriter out = new LightWriter(outputStream);
        No1066RedAndBlueAndMoreVariousColorsEasy solver = new No1066RedAndBlueAndMoreVariousColorsEasy();
        solver.solve(1, in, out);
        out.close();
    }

    static class No1066RedAndBlueAndMoreVariousColorsEasy {
        private static final int MOD = 998244353;

        public void solve(int testNumber, LightScanner in, LightWriter out) {
            int n = in.ints(), q = in.ints();
            long[][] dp = new long[2][n + 1];
            dp[1][0] = 1;
            for (int i = 0; i < n; i++) {
                int a = in.ints();
                System.arraycopy(dp[1], 0, dp[0], 0, n + 1);
                for (int j = 0; j <= n; j++) {
                    dp[1][j] *= (a - 1);
                    if (0 < j) dp[1][j] += dp[0][j - 1];
                    dp[1][j] %= MOD;
                }
            }
            for (int i = 0; i < q; i++) out.ans(dp[1][in.ints()]).ln();
        }

    }

    static class LightWriter implements AutoCloseable {
        private final Writer out;
        private boolean autoflush = false;
        private boolean breaked = true;

        public LightWriter(Writer out) {
            this.out = out;
        }

        public LightWriter(OutputStream out) {
            this(new OutputStreamWriter(new BufferedOutputStream(out), Charset.defaultCharset()));
        }

        public LightWriter print(char c) {
            try {
                out.write(c);
                breaked = false;
            } catch (IOException ex) {
                throw new UncheckedIOException(ex);
            }
            return this;
        }

        public LightWriter print(String s) {
            try {
                out.write(s, 0, s.length());
                breaked = false;
            } catch (IOException ex) {
                throw new UncheckedIOException(ex);
            }
            return this;
        }

        public LightWriter ans(String s) {
            if (!breaked) {
                print(' ');
            }
            return print(s);
        }

        public LightWriter ans(long l) {
            return ans(Long.toString(l));
        }

        public LightWriter ln() {
            print(System.lineSeparator());
            breaked = true;
            if (autoflush) {
                try {
                    out.flush();
                } catch (IOException ex) {
                    throw new UncheckedIOException(ex);
                }
            }
            return this;
        }

        public void close() {
            try {
                out.close();
            } catch (IOException ex) {
                throw new UncheckedIOException(ex);
            }
        }

    }

    static class LightScanner implements AutoCloseable {
        private BufferedReader reader = null;
        private StringTokenizer tokenizer = null;

        public LightScanner(InputStream in) {
            reader = new BufferedReader(new InputStreamReader(in));
        }

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

        public int ints() {
            return Integer.parseInt(string());
        }

        public void close() {
            try {
                this.reader.close();
            } catch (IOException ex) {
                throw new UncheckedIOException(ex);
            }
        }

    }
}
0