結果

問題 No.3266 岩井星人は見ずにはいられない
ユーザー deuteridayo
提出日時 2025-09-06 15:44:53
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,815 bytes
コンパイル時間 6,379 ms
コンパイル使用メモリ 334,912 KB
実行使用メモリ 488,768 KB
最終ジャッジ日時 2025-09-06 15:45:09
合計ジャッジ時間 12,941 ms
ジャッジサーバーID
(参考情報)
judge / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample TLE * 1 -- * 3
other -- * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

#ifndef MY_HEADER
#define MY_HEADER

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using lint = long long;
using ulint = unsigned long long;
using llint = __int128_t;
struct edge;
using graph = vector<vector<edge>>;
#define endl '\n'
constexpr int INF = 1<<30;
constexpr lint INF64 = 1LL<<61;
constexpr lint mod107 = 1e9+7;
using mint107 = modint1000000007;
constexpr long mod = 998244353;
using mint = modint998244353;
lint ceilDiv(lint x, lint y){if(x >= 0){return (x+y-1)/y;}else{return x/y;}}
lint floorDiv(lint x, lint y){if(x >= 0){return x/y;}else{return (x-y+1)/y;}}
lint Sqrt(lint x) {assert(x >= 0); lint ans = sqrt(x); while(ans*ans > x)ans--; while((ans+1)*(ans+1)<=x)ans++; return ans;}
lint gcd(lint a,lint b){if(a<b)swap(a,b);if(a%b==0)return b;else return gcd(b,a%b);}
lint lcm(lint a,lint b){return (a / gcd(a,b)) * b;}
double Dist(double x1, double y1, double x2, double y2){return sqrt(pow(x1-x2, 2) + pow(y1-y2,2));}
lint DistSqr(lint x1, lint y1, lint x2, lint y2){return (x1-x2)*(x1-x2) + (y1-y2)*(y1-y2); }
string toString(lint n){string ans = "";if(n == 0){ans += "0";}else{while(n > 0){int a = n%10;char b = '0' + a;string c = "";c += b;n /= 10;ans = c + ans;}}return ans;}
string toString(lint n, lint k){string ans = toString(n);string tmp = "";while(ans.length() + tmp.length() < k){tmp += "0";}return tmp + ans;}
vector<lint>prime;void makePrime(lint n){prime.push_back(2);for(lint i=3;i<=n;i+=2){bool chk = true;for(lint j=0;j<prime.size() && prime[j]*prime[j] <= i;j++){if(i % prime[j]==0){chk=false;break;}}if(chk)prime.push_back(i);}}
lint Kai[20000001]; bool firstCallnCr = true; 
lint ncrmodp(lint n,lint r,lint p){ if(firstCallnCr){ Kai[0] = 1; for(int i=1;i<=20000000;i++){ Kai[i] = Kai[i-1] * i; Kai[i] %= p;} firstCallnCr = false;} if(n<0)return 0; if(r<0)return 0;
if(n < r)return 0;if(n==0)return 1;lint ans = Kai[n];lint tmp = (Kai[r] * Kai[n-r]) % p;for(lint i=1;i<=p-2;i*=2){if(i & p-2){ans *= tmp;ans %= p;}tmp *= tmp;tmp %= p;}return ans;}
#define rep(i, n) for(int i = 0; i < n; i++)
#define repp(i, x, y) for(int i = x; i < y; i++)
#define rrep(i, x) for(int i = x-1; i >= 0; i--)
#define vec vector
#define pb push_back
#define eb emplace_back
#define se second
#define fi first
#define al(x) x.begin(),x.end()
#define ral(x) x.rbegin(),x.rend()

struct edge{
    edge(lint v, lint c = 1) {to = v, cost = c;}
    lint to;
    lint cost;
};

#endif

lint N, A;
string S;

map<pair<lint, lint>, pair<lint, lint>>dp;
map<pair<lint, lint>, bool>went;
// nextrate, account
pair<lint, lint> f(int rate, int step) {
    if(went[{rate,step}]) return dp[{rate, step}];
    went[{rate, step}] = true;
    if(step == 0) {
        int r = rate;
        int account = 0;
        rep(i, N) {
            if(S[i] == '1') {
                if(r != 1200) {
                    r++;
                    account++;
                }
            } else {
                r--;
            }
        }
        return dp[{rate, step}] = {r, account};
    } else {
        auto p = f(rate, step-1);
        auto p2 = f(p.first, step-1);
        return dp[{rate, step}] = {p2.first, min(p.second + p2.second, INF64)};
    }

}


int main(){
    cin >> N >> A;
    cin >> S;
    lint ans = 0;
    lint ac = 0;
    int rate = 1200;
    for(int i = 60; i >= 0; i--) {
        auto p = f(rate, i);
        if(ac + p.second < A) {
            rate = p.first;
            ac += p.second;
            ans += (1LL<<i) * N;
        }
    }

    rep(i, N) {


        if(S[i] == '1') {
            if(rate != 1200) {
                rate++;
                ac++;
            }
        } else {
            rate--;
        }
        
        ans++;
        if(ac == A) {
            cout << ans << endl;
            return 0;
        }



    }



    
}

0