結果

問題 No.2171 OR Assignment
ユーザー hotman78hotman78
提出日時 2022-12-23 07:47:47
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,393 bytes
コンパイル時間 2,258 ms
コンパイル使用メモリ 215,312 KB
最終ジャッジ日時 2025-02-09 19:00:58
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12 WA * 8 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma optimize("Ofast")
#pragma target("avx2")
#include<bits/stdc++.h>
using namespace std;
#include<atcoder/modint.hpp>
using mint=atcoder::static_modint<998244353>;
#define endl '\n'
using lint=long long;
#define all(n) (n).begin(),(n).end()

int main(){
    cin.tie(0)->sync_with_stdio(0);
    int n;
    cin>>n;
    vector<int>a(n);
    for(int i=0;i<n;++i)cin>>a[i];
    vector<vector<int>>s(n+1);
    s[0].emplace_back(0);
    for(int i=0;i<n;++i){
        for(auto e:s[i]){
            s[i+1].emplace_back(e|a[i]);
        }
        s[i+1].emplace_back(a[i]);
        s[i+1].emplace_back(0);
        sort(all(s[i+1]));
        s[i+1].erase(unique(all(s[i+1])),s[i+1].end());
    }
    mint ans=0;
    vector<lint>dp={1};
    for(int i=0;i<n;++i){
        vector<pair<int,int>>g;
        for(auto d:s[i]){
            for(auto e:s[i]){
                if(d<e)break;
                if((d|e)==d){
                    g.emplace_back(d,e|a[i]);
                }
            }
        }
        sort(all(g));
        g.erase(unique(all(g)),g.end());
        vector<lint>dp2(s[i+1].size());
        auto get=[&](const vector<int>&x,int i){
            return lower_bound(all(x),i)-x.begin();
        };
        for(auto [p,q]:g){
            dp2[get(s[i+1],q)]+=dp[get(s[i],p)];
            if(i+1==n)ans+=dp[get(s[i],p)];
        }
        swap(dp,dp2);
    }
    cout<<ans.val()<<endl;
}
0