結果

問題 No.1654 Binary Compression
ユーザー pockyny
提出日時 2021-08-21 04:14:29
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,225 bytes
コンパイル時間 991 ms
コンパイル使用メモリ 79,060 KB
最終ジャッジ日時 2025-01-24 01:10:00
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>

using namespace std;
typedef long long ll;
ll mod = 924844033,dp[3000010] = {};
vector<ll> v,u;
int main(){
    string s; cin >> s;
    int i,n = s.size();
    v.push_back(-1);
    for(i=0;i<n;i++){
        if(s[i]=='1') v.push_back(i);
    }
    v.push_back(n);
    for(i=0;i<(int)v.size() - 1;i++) u.push_back(v[i + 1] - v[i] - 1);
    if(u.size()==1){
        cout << u[0] + 1 << endl;
        return 0;
    }
    ll ans = (u[0] + 1LL)*(u.back() + 1LL)%mod;
    if(u.size()==2){
        cout << ans << endl;
        return 0;
    }
    u.erase(u.begin()); u.pop_back();
    vector<pair<int,int>> st;
    st.push_back({(int)u.size(),1000000000});
    vector<int> r(u.size());
    for(i=u.size() - 1;i>=0;i--){
        while(st.back().second<=u[i]) st.pop_back();
        r[i] = st.back().first;
        st.push_back({i,u[i]});
    }
    dp[0] += u[0] + 1; dp[r[0] - 1] += 1;
    for(i=1;i<u.size();i++){
        (dp[i] += (u[i] + 1)*dp[i - 1]%mod) %= mod;
        (dp[r[i] - 1] += dp[i - 1]) %= mod;
    }
    /*for(i=0;i<u.size();i++) cout << u[i] << " ";
    cout << endl;
    cout << ans << endl;*/
    (ans *= dp[u.size() - 1]) %= mod;
    cout << ans << endl;
}
0