Submission #62489

ソースコード

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#define ll long long
#define rep(i,n) for(ll i=0;i<(n);i++)
#define pll pair<ll,ll>
#define pii pair<int,int>
#define pq priority_queue
#define pb push_back
#define eb emplace_back
#define fi first
#define se second
#define endl '\n'
#define ios ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define lb(c,x) distance(c.begin(),lower_bound(all(c),x))
#define ub(c,x) distance(c.begin(),upper_bound(all(c),x))
using namespace std;
template<class T> inline bool chmax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
template<class T> inline bool chmin(T& a,T b){if(a>b){a=b;return 1;}return 0;}
const ll INF=1e9+7;
const int mod = 998244353;
struct mint {
ll x; // typedef long long ll;
mint(ll x=0):x((x%mod+mod)%mod){}
mint operator-() const { return mint(-x);}
mint& operator+=(const mint a) {
if ((x += a.x) >= mod) x -= mod;
return *this;
}
mint& operator-=(const mint a) {
if ((x += mod-a.x) >= mod) x -= mod;
return *this;
}
mint& operator*=(const mint a) { (x *= a.x) %= mod; return *this;}
mint operator+(const mint a) const { return mint(*this) += a;}
mint operator-(const mint a) const { return mint(*this) -= a;}
mint operator*(const mint a) const { return mint(*this) *= a;}
mint pow(ll t) const {
if (!t) return 1;
mint a = pow(t>>1);
a *= a;
if (t&1) a *= *this;
return a;
}
// for prime mod
mint inv() const { return pow(mod-2);}
mint& operator/=(const mint a) { return *this *= a.inv();}
mint operator/(const mint a) const { return mint(*this) /= a;}
};
istream& operator>>(istream& is, mint& a) { return is >> a.x;}
ostream& operator<<(ostream& os, const mint& a) { return os << a.x;}
int main(){
ios;
int n;
cin >> n;
vector<ll> a(n),p(n);
mint v=2;
rep(i,n){
cin >> a[i];
p[i]=(v.pow(a[i]).x+mod-1)%mod;
}
vector<ll> dp1(n+1),dp2(n+1);
dp1[0]=1,dp2[0]=1;
rep(i,n){ //i番目まで見たとき、j種類選ぶ方法
for(int j=1;j<=n;j++){
dp1[j]=(dp2[j]+(dp2[j-1]*p[i])%mod)%mod;
}
dp2=dp1;
}
rep(i,n){
dp2[i+1]+=dp2[i];
dp2[i+1]%=mod;
}
int q;
cin >> q;
rep(i,q){
int l,r;
cin >> l >> r;
l--;
cout << (dp2[r]-dp2[l]+mod)%mod << endl;
}
return 0;
}

ステータス

項目 データ
問題 1351 - Colorful Balls
ユーザー名 unos
投稿日時 2020-08-18 21:30:42
言語 C++17
状態 Accepted
得点 100
ソースコード長 2463 Byte
最大実行時間 239 ms
最大メモリ使用量 64536 KB

セット

セット 得点 Cases
1 ALL 100 / 100 *

テストケース

ファイル名 状態 実行時間 メモリ使用量 #
input01 AC 22 ms 604 KB
1
input02 AC 16 ms 560 KB
1
input03 AC 16 ms 388 KB
1
input04 AC 19 ms 476 KB
1
input05 AC 21 ms 560 KB
1
input06 AC 14 ms 516 KB
1
input07 AC 25 ms 596 KB
1
input08 AC 16 ms 680 KB
1
input09 AC 29 ms 628 KB
1
input10 AC 17 ms 676 KB
1
input11 AC 50 ms 2656 KB
1
input12 AC 123 ms 9632 KB
1
input13 AC 144 ms 15716 KB
1
input14 AC 134 ms 19528 KB
1
input15 AC 36 ms 19836 KB
1
input16 AC 108 ms 21384 KB
1
input17 AC 215 ms 29624 KB
1
input18 AC 239 ms 39268 KB
1
input19 AC 215 ms 48916 KB
1
input20 AC 155 ms 57576 KB
1
input21 AC 99 ms 59060 KB
1
input22 AC 80 ms 59236 KB
1
input23 AC 111 ms 64536 KB
1
input_sample01 AC 21 ms 64436 KB
1
input_sample02 AC 16 ms 64392 KB
1