結果

問題 No.1654 Binary Compression
ユーザー ygussanyygussany
提出日時 2021-06-17 12:18:58
言語 C
(gcc 12.3.0)
結果
WA  
実行時間 -
コード長 2,348 bytes
コンパイル時間 310 ms
コンパイル使用メモリ 33,408 KB
実行使用メモリ 201,344 KB
最終ジャッジ日時 2024-10-14 02:12:27
合計ジャッジ時間 5,351 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 WA -
testcase_03 AC 1 ms
5,248 KB
testcase_04 AC 1 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 WA -
testcase_34 RE -
testcase_35 WA -
testcase_36 WA -
testcase_37 RE -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 AC 166 ms
201,344 KB
testcase_43 WA -
testcase_44 AC 14 ms
5,248 KB
testcase_45 AC 15 ms
5,248 KB
testcase_46 AC 161 ms
201,216 KB
testcase_47 AC 2 ms
5,248 KB
testcase_48 AC 1 ms
5,248 KB
testcase_49 WA -
testcase_50 RE -
testcase_51 WA -
testcase_52 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <stdlib.h>

const int Mod = 998244353;

typedef struct {
	int left, right;
	long long sum, lx;
} lazy_seg_node;

void init_node(lazy_seg_node v[], int k, int l, int r)
{
	v[k].left = l;
	v[k].right = r;
	v[k].sum = 0;
	v[k].lx = -1;
	if (l < r) {
		init_node(v, k << 1, l, (l + r) / 2);
		init_node(v, (k << 1) ^ 1, (l + r) / 2 + 1, r);
	}
}

void update_segment(lazy_seg_node v[], int k, int l, int r, long long x)
{
	int i, j;
	if (v[k].left > r || v[k].right < l) return;
	else if (v[k].left >= l && v[k].right <= r) {
		v[k].lx = x;
		v[k].sum = (x * (v[k].right - v[k].left + 1)) % Mod;
		for (j = k, i = j >> 1; i > 0; j = i, i >>= 1) v[i].sum = (v[j].sum + v[j^1].sum) % Mod;
	} else {
		if (v[k].lx >= 0) {
			j = k << 1;
			v[j].sum = (v[k].lx * (v[j].right - v[j].left + 1)) % Mod;
			v[j^1].sum = (v[k].lx * (v[j^1].right - v[j^1].left + 1)) % Mod;
			v[j].lx = v[k].lx;
			v[j^1].lx = v[k].lx;
			v[k].lx = -1;
		}
		update_segment(v, k << 1, l, r, x);
		update_segment(v, (k << 1) ^ 1, l, r, x);
	}
}

long long get_sum(lazy_seg_node v[], int k, int l, int r)
{
	int j;
	if (v[k].left > r || v[k].right < l) return 0;
	else if (v[k].left >= l && v[k].right <= r) return v[k].sum;
	else {
		if (v[k].lx >= 0) {
			j = k << 1;
			v[j].sum = (v[k].lx * (v[j].right - v[j].left + 1)) % Mod;
			v[j^1].sum = (v[k].lx * (v[j^1].right - v[j^1].left + 1)) % Mod;
			v[j].lx = v[k].lx;
			v[j^1].lx = v[k].lx;
			v[k].lx = -1;
		}
		return (get_sum(v, k << 1, l, r) + get_sum(v, (k << 1) ^ 1, l, r)) % Mod;
	}
}

int main()
{
	char S[2000001];
	scanf("%s", S);
	
	int i, j, k = 0, l, n = 0, A[2000001];
	lazy_seg_node *v;
	for (i = 0; S[i] != 0; i = j) {
		for (j = i + 1; S[j] == S[i]; j++);
		if (S[i] == '0') {
			if (n < j - i) n = j - i;
			A[k++] = i - j;
		} else A[k++] = j - i;
	}
	l = i;
	v = (lazy_seg_node*)malloc(sizeof(lazy_seg_node) * (n + 1) * 3);
	init_node(v, 1, 1, n);
	
	long long ans = 1, zero_lr = 1;
	if (A[k-1] < 0) zero_lr = -A[--k] + 1;
	if (k > 0) {
		if (A[0] < 0) zero_lr = zero_lr * (-A[0] + 1) % Mod;
		ans = A[k-1];
		for (i = k - 3, j = k - 2; i >= 0; i -= 2, j -= 2) {
			update_segment(v, 1, 1, -A[j], ans);
			ans += (get_sum(v, 1, 1, n) + 1) * A[i] % Mod;
			if (ans >= Mod) ans -= Mod;
		}
	} else zero_lr--;
	printf("%lld\n", ans * zero_lr % Mod);
	fflush(stdout);
	return 0;
}
0