結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
ユーザー rtia_iidxrtia_iidx
提出日時 2020-05-29 22:21:58
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 6,794 bytes
コンパイル時間 1,086 ms
コンパイル使用メモリ 105,540 KB
実行使用メモリ 284,824 KB
最終ジャッジ日時 2023-08-06 02:37:31
合計ジャッジ時間 4,317 ms
ジャッジサーバーID
(参考情報)
judge12 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,508 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 1 ms
4,380 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 149 ms
104,872 KB
testcase_09 AC 2 ms
4,376 KB
testcase_10 MLE -
testcase_11 AC 81 ms
60,068 KB
testcase_12 AC 62 ms
46,076 KB
testcase_13 AC 85 ms
61,932 KB
testcase_14 AC 91 ms
65,360 KB
testcase_15 MLE -
testcase_16 AC 37 ms
27,544 KB
testcase_17 AC 42 ms
32,032 KB
testcase_18 MLE -
testcase_19 AC 80 ms
57,952 KB
testcase_20 AC 2 ms
4,380 KB
testcase_21 MLE -
testcase_22 AC 2 ms
4,376 KB
testcase_23 AC 1 ms
4,376 KB
testcase_24 AC 1 ms
4,376 KB
testcase_25 MLE -
testcase_26 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<vector>
#include<algorithm>
#include<functional>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<unordered_map>
#include<climits>
#include<cstdlib>
#include<cmath>
#include<string>
#include<iomanip>
#include<bitset>
#include<list>

/*
                     __,.二ニ==-  ニ.、_.                     
                .,..‐ ⌒            ``'ァ-ニ、.               
              ィ'´.                      ´丶.、            
           .,ィ'´                            `.x、..         
 ..        /.                                、\.        
        .ン′.                                  ¬`""冖ーミト.
       ,r′.                                     .ヘ、    `
      ツ                                         `、....   
     ./´                                          ヘ    
 .....  /.                                            .〉.   
   ´/..                                             .l   
   ィ                         .                     f..  
 .. .f ,d.                 l   ′ 」  ,.             ト     !  
 . 〕../.f..                 ′ ..  |  .}             |      |.  
  .!./..f..                 /  !- ナ丶п冖т    ノー- . 〕      |.  
  |メ |               | j      ,  ┌. |〈.   л`.   /|..     ┤,.. 
 ...「...|.      |  ´      l. |  j.L......ュ.L_└ヽ_|Y.  メムw ょ |  j.:   |  ̄
 .  |.               т〕<.ィ冖T冖.. г‐ `、 `, /┴¬..г ̄|.. .′  |  
 ...  |        ...       ),|.. `  リ 「_ノ.||    ` V  |!{,「ll ´.」.     卜  
 .  |.」      ′        ヽ    └++〃..        ルwf  カz′.   |.  
    |..〕                                      「     .|   
    .l.|.                                      ′.     |   
 .   .〕..               `!        _.....ー:''         」 ´    λ.  
    _「.    ,            ┐_,、`~‐''"´                ィ   .、 ヘ、 
    f      :__..,二ュ.-i―'''^~´                       、\イ ヘ.`x
 .  /      {                                     j    .~^
  、/       't..                                   丿..     
 .../.       ,x┐..                                 ∠∫     
 :^.       /   ̄冖ー=zzュ┌ー―--   ∟,二..._.       _,、.-ー.'l+~. .l`.    
 .       У.         ⌒冖‐-=._.. l「.「      ´ ̄」了  .,、-''^   〉 ヽ_   
       _/.                   ̄~'.ー-=.、_,..usァ.ー''"      {   \´ 
      _ヰl'¬―- 、_                                 (    .\
*/

using namespace std;

using ll = long long int;
using ull = unsigned long long int;

ll const MOD = 998244353;
ll const INF = (long long int)1 << 61;

ll mypow(ll x,ll n,ll mod = MOD){
    ll ret = 1;
    while(n > 0){
        if(n&1){
            ret = (ret*x)%mod;
        }
        x = (x*x)%mod;
        n >>= 1;
    }
    return ret;
}

ll mygcd(ll a,ll b){
    if(b == 0)return a;
    return mygcd(b,a%b);
}

ll twoPow(ll shiftNum){
    return (1LL << (shiftNum - 1));
}

int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
 
    ll n,q;
    cin >> n >> q;

    vector<ll> a(n);
    vector<ll> b(q);

    for(int i = 0; i < n; i++){
        cin >> a[i];
    }

    for(int i = 0; i < q; i++){
        cin >> b[i];
    }

    vector<ll> p(n);
    vector<ll> rp(n);
    ll total = 1;

    for(int i = 0; i < n; i++){
        p[i] = mypow(a[i],MOD - 2);
        rp[i] = (1 + (MOD - p[i]))%MOD;
        total = (total * a[i])%MOD;
    }

    vector<vector<ll>> dp(n+1,vector<ll>(n+1,0));

    dp[0][0] = 1;

    for(int i = 0; i < n; i++){
        for(int j = 0; j < n+1; j++){
            dp[i+1][j] += (dp[i][j]*rp[i])%MOD;
            dp[i+1][j] %= MOD;
            if(j+1 < n+1){
                dp[i+1][j+1] += (dp[i][j]*p[i])%MOD;
                dp[i+1][j+1] %= MOD;
            }
        }
    }

    for(int i = 0; i < q; i++){
        cout << (dp[n][b[i]]*total)%MOD << endl;
    }

    return 0;
}
0