結果
問題 | No.1741 Arrays and XOR Procedure |
ユーザー |
![]() |
提出日時 | 2023-03-16 13:18:01 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 55 ms / 2,000 ms |
コード長 | 2,451 bytes |
コンパイル時間 | 2,841 ms |
コンパイル使用メモリ | 250,336 KB |
実行使用メモリ | 15,088 KB |
最終ジャッジ日時 | 2024-09-18 09:19:48 |
合計ジャッジ時間 | 5,247 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include<bits/stdc++.h>using namespace std;using ll = long long;const ll mod = 998'244'353;//const ll mod = 1'000'000'007;//const ll mod = 67'280'421'310'721;struct mint{long long x;mint(long long 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{mint res(*this);return res+=a;}mint operator-(const mint& a) const{mint res(*this);return res-=a;}mint operator*(const mint& a) const{mint res(*this);return res*=a;}mint pow(long long n) const {assert(0 <= n);mint a = *this, r = 1;while (n) {if (n & 1) r *= a;a *= a;n >>= 1;}return r;}mint inv() const{return pow(mod-2);}mint& operator/=(const mint& a){return (*this)*=a.inv();}mint operator/(const mint& a) const {mint res(*this);return res/=a;}friend ostream& operator<<(ostream& os, const mint& m){os << m.x;return os;}bool operator==(const mint& a) const {return x == a.x;}bool operator<(const mint& a) const{return x < a.x;}};int main(){int n;cin>>n;vector<int> a(n);for(int i = 0;i<n;i++) cin>>a[i];vector<vector<mint > > dp(n,vector<mint>(2,0));if(a[0]==0) dp[0][0] += 1;else if(a[0]==1) dp[0][1] += 1;else{dp[0][0] += 1;dp[0][1] += 1;}for(int i = 1;i<n;i++){if((i&(n-1))==i){if(a[i]==-1){dp[i][0] = dp[i-1][0] + dp[i-1][1];dp[i][1] = dp[i-1][0] + dp[i-1][1];}else{dp[i][0] = dp[i-1][0^a[i]];dp[i][1] = dp[i-1][1^a[i]];}}else{if(a[i]==-1){dp[i][0] = mint(2) * dp[i-1][0];dp[i][1] = mint(2) * dp[i-1][1];}else{dp[i][0] = dp[i-1][0];dp[i][1] = dp[i-1][1];}}}cout<<dp[n-1][1]<<endl;}