結果

問題 No.3266 岩井星人は見ずにはいられない
ユーザー shobonvip
提出日時 2025-09-07 10:22:39
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 1,073 bytes
コンパイル時間 2,165 ms
コンパイル使用メモリ 198,604 KB
実行使用メモリ 8,320 KB
最終ジャッジ日時 2025-09-07 10:22:43
合計ジャッジ時間 3,589 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

/**
	author:  shobonvip
	created: 2025.09.07 09:11:53
**/

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

struct S {
	ll sum, mx;
};

S op(S a, S b) {
	return S{a.sum + b.sum, max(a.mx, a.sum + b.mx)};
}

S e() {
	return S{0, 0};
}

int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);

	ll n, a; cin >> n >> a;
	string s; cin >> s;
	vector<S> rui(n + 1);
	vector<ll> rui_1num(n + 1);
	rui[0] = e();
	for (int i=0; i<n; i++) {
		if (s[i] == '0') {
			rui[i+1] = op(rui[i], S{-1, 0});
		} else {
			rui[i+1] = op(rui[i], S{1, 1});
			rui_1num[i+1] += 1;
		}
		rui_1num[i+1] += rui_1num[i];
	}

	auto check = [&](ll t) -> ll {
		ll p = t / n;
		ll q = t % n;
		S v = e();
		ll nums = rui_1num[n] * p + rui_1num[q];
		{
			S w = rui[n];
			while (p > 0) {
				if (p & 1) v = op(v, w);
				w = op(w, w);
				p >>= 1;
			}
		}
		v = op(v, rui[q]);
		return nums - v.mx;
	};

	ll ub = 2e18; // OK
	ll lb = 0; // NG
	while (ub - lb > 1) {
		ll t = (ub + lb) / 2;
		if (check(t) >= a) {
			ub = t;
		}else {
			lb = t;
		}
	}
	cout << ub << endl;
}

0