結果

問題 No.2281 K → K-1 01 Flip
ユーザー magsta
提出日時 2023-04-19 14:58:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,155 bytes
コンパイル時間 1,636 ms
コンパイル使用メモリ 169,944 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-10-15 10:38:25
合計ジャッジ時間 6,590 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other TLE * 1 -- * 55
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
	long long N, Q; cin >> N >> Q;
	string S; cin >> S;

	vector<long long> rui_l(N, 0), rui_r(N, 0);
	for (int i = 0; i < N; i++) {
		if (i != 0) rui_l[i] = rui_l[i - 1];
		if (S[i] == '0') rui_l[i] += 1;
		else rui_l[i] -= 1;
		if (i != 0) rui_r[N - i - 1] = rui_r[N - i];
		if (S[N - i - 1] == '0') rui_r[N - i - 1] += 1;
		else rui_r[N - i - 1] -= 1;
	}

	for (int i = 0; i < Q; i++) {
		long long L, R, K; cin >> L >> R >> K;
		long long now = 0;
		bool flag = false;
		for (int i = L - 1; i < R; i++) {
			if (S[i] == '0') {
				now++;
				if (now >= K) flag = true;
			}
			else now = 0;
		}
		now = 0;
		for (int i = 0; i < N; i++) {
			if (S[i] == '1') {
				now++;
				if (now >= K) flag = true;
			}
			else now = 0;
		}

		if (!flag) cout << R - L + 1 << endl;
		else {
			long long sum = rui_l[N - 1];
			if (L != 1) sum -= rui_l[L - 2];
			if (R != N) sum -= rui_r[R];
			long long amari = (sum + 1000000 * (2 * K - 1)) % (2 * K - 1);
			if (amari == 0) cout << 2 * (K - 1) << endl;
			else if (amari < K) cout << 2 * K - 2 - amari << endl;
			else cout << amari - 1 << endl;
		}
	}
}
0