結果

問題 No.1654 Binary Compression
ユーザー ygussanyygussany
提出日時 2021-06-17 12:23:20
言語 C
(gcc 12.3.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 2,411 bytes
コンパイル時間 345 ms
コンパイル使用メモリ 32,768 KB
実行使用メモリ 11,520 KB
最終ジャッジ日時 2024-10-14 02:13:02
合計ジャッジ時間 3,328 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
testcase_47 AC 1 ms
5,248 KB
testcase_48 AC 1 ms
5,248 KB
testcase_49 RE -
testcase_50 RE -
testcase_51 RE -
testcase_52 RE -
権限があれば一括ダウンロードができます

ソースコード

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, n = 0, A[2000001];
	lazy_seg_node *v;
	for (i = 0; S[i] != 0; i = j) {
		if (S[i] != '0' && S[i] != '1') return -1;
		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;
	}
	if (i > 2000000) return -1;
	v = (lazy_seg_node*)malloc(sizeof(lazy_seg_node) * (n + 1) * 4);
	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