結果

問題 No.3266 岩井星人は見ずにはいられない
ユーザー edon8618
提出日時 2025-09-06 15:23:19
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 12 ms / 2,000 ms
コード長 3,675 bytes
コンパイル時間 6,955 ms
コンパイル使用メモリ 334,196 KB
実行使用メモリ 7,716 KB
最終ジャッジ日時 2025-09-06 15:23:28
合計ジャッジ時間 6,580 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;

// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;

// # pragma GCC target("avx2")
// # pragma GCC optimize("O3")
// # pragma GCC optimize("unroll-loops")

#define ll long long
#define rep(i,n) for(ll i=0;i<(ll)(n);i++)
#define vi vector<int>
#define vl vector<ll>
#define vd vector<double>
#define vb vector<bool>
#define vs vector<string>
#define vc vector<char>
#define ull unsigned long long
#define chmax(a,b) a=max(a,b)
#define chmin(a,b) a=min(a,b)



ll inf=(1ll<<60);
// const double PI=3.1415926535897932384626433832795028841971;


// ll rui(ll a,ll b){
//     if(b==0)return 1;
//     if(b%2==1) return a*rui(a*a,b/2);
//     return rui(a*a,b/2);
// }

// ll kai(ll n){
//     if(n==0)return 1;
//     return n*kai(n-1);
// }

// using mint = modint998244353;//static_modint<998244353>
// using mint = modint1000000007;//static_modint<1000000007>
// using mint = modint;//mint::set_mod(mod);


// ll const mod=1000000007ll;
// ll const mod=998244353ll;
// ll modrui(ll a,ll b,ll mod){
//     a%=mod;
//     if(b==0)return 1;
//     if(b%2==1) return a*modrui(a*a%mod,b/2,mod)%mod;
//     return modrui(a*a%mod,b/2,mod)%mod;
// }

// ll inv(ll x){
//     x%=mod;
//     return modrui(x,mod-2);
// }

// ll modkai(ll n){
//     ll ret=1;
//     rep(i,n){
//         ret*=(i+1)%mod;
//         ret%=mod;
//     }
//     return ret;
// }


// void incr(vl &v,ll n){
//     ll k=v.size();
//     v[k-1]++;
//     ll now=k-1;
//     while (v[now]>=n)
//     {
//         v[now]=0;
//         if(now==0)break;
//         v[now-1]++;
//         now--;
//     }
//     return;
// }










void solve(){
    ll n,a,r=1200;
    cin >> n >> a;
    string s;
    cin >> s;
    ll ans=0;
    unordered_map<ll,ll> mp;
    bool fl=1;
    while(fl){
        fl=0;
        ll cnt=0;
        ll init=r;
        rep(i,n){
            ans++;
            if(s[i]=='1'){
                if(r<1200){
                    r++;
                    a--;
                    cnt++;
                    if(a==0){
                        cout << ans << endl;
                        return;
                    }
                }
                else{
                    fl=1;
                }
            }
            else{
                r--;
            }
        }
        if(mp.find(init)==mp.end())mp[init]=cnt;
        else{
            break;
        }
    }
    
    {
        ll x=0;
        {
            ll z=r;
            rep(i,n){
                if(s[i]=='1' && z<1200){
                    x++;
                    z++;
                }
                if(s[i]=='0')z--;
            }
        }
        ans+=(a/x)*n;
        a%=x;
        if(a==0){
            a=x;
            ans-=n;
        }
        rep(i,n){
            ans++;
            if(s[i]=='1' && r<1200){
                a--;
                r++;
            }
            if(s[i]=='0')r--;
            if(a==0){
                cout << ans << endl;
                return;
            }
        }
    }
    // else{
    //     ans+=(a/mp[r])*n;
    //     a%=mp[r];
    //     if(a==0){
    //         a=mp[r]
    //     }
    //     rep(i,n){
    //         ans++;
    //         if(s[i]=='1' && r<1200){
    //             a--;
    //             r++;
    //         }
    //         if(s[i]=='0')r--;
    //         if(a==0){
    //             cout << ans << endl;
    //             return;
    //         }
    //     }
    // }

}

int main(){
    // ios::sync_with_stdio(false);
    // std::cin.tie(nullptr);
    ll t=1;
    // cin >> t;
    while(t--)solve();
}
0