結果

問題 No.1720 Division Permutation
ユーザー 👑 ygussanyygussany
提出日時 2021-10-16 15:59:31
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 299 ms / 4,000 ms
コード長 5,937 bytes
コンパイル時間 1,027 ms
コンパイル使用メモリ 37,188 KB
実行使用メモリ 67,192 KB
最終ジャッジ日時 2023-10-24 11:34:29
合計ジャッジ時間 13,694 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 15 ms
55,216 KB
testcase_01 AC 15 ms
55,216 KB
testcase_02 AC 14 ms
55,216 KB
testcase_03 AC 14 ms
55,216 KB
testcase_04 AC 14 ms
55,216 KB
testcase_05 AC 15 ms
55,216 KB
testcase_06 AC 15 ms
55,216 KB
testcase_07 AC 14 ms
55,216 KB
testcase_08 AC 13 ms
55,216 KB
testcase_09 AC 13 ms
55,216 KB
testcase_10 AC 14 ms
55,216 KB
testcase_11 AC 14 ms
55,216 KB
testcase_12 AC 15 ms
55,216 KB
testcase_13 AC 232 ms
65,632 KB
testcase_14 AC 232 ms
65,632 KB
testcase_15 AC 147 ms
64,876 KB
testcase_16 AC 224 ms
65,576 KB
testcase_17 AC 157 ms
64,976 KB
testcase_18 AC 198 ms
67,192 KB
testcase_19 AC 291 ms
65,756 KB
testcase_20 AC 286 ms
65,704 KB
testcase_21 AC 290 ms
65,696 KB
testcase_22 AC 286 ms
65,680 KB
testcase_23 AC 286 ms
65,896 KB
testcase_24 AC 286 ms
65,708 KB
testcase_25 AC 284 ms
65,700 KB
testcase_26 AC 285 ms
65,684 KB
testcase_27 AC 286 ms
65,692 KB
testcase_28 AC 281 ms
65,728 KB
testcase_29 AC 299 ms
65,904 KB
testcase_30 AC 296 ms
65,764 KB
testcase_31 AC 286 ms
65,684 KB
testcase_32 AC 14 ms
55,216 KB
testcase_33 AC 14 ms
55,216 KB
testcase_34 AC 14 ms
55,216 KB
testcase_35 AC 14 ms
55,216 KB
testcase_36 AC 14 ms
55,216 KB
testcase_37 AC 14 ms
55,216 KB
testcase_38 AC 15 ms
55,216 KB
testcase_39 AC 14 ms
55,216 KB
testcase_40 AC 14 ms
55,216 KB
testcase_41 AC 14 ms
55,216 KB
testcase_42 AC 14 ms
55,216 KB
testcase_43 AC 200 ms
65,412 KB
testcase_44 AC 185 ms
65,268 KB
testcase_45 AC 169 ms
65,084 KB
testcase_46 AC 129 ms
64,700 KB
testcase_47 AC 173 ms
65,144 KB
testcase_48 AC 139 ms
64,792 KB
testcase_49 AC 207 ms
65,464 KB
testcase_50 AC 165 ms
65,072 KB
testcase_51 AC 138 ms
64,784 KB
testcase_52 AC 188 ms
65,300 KB
testcase_53 AC 231 ms
65,640 KB
testcase_54 AC 231 ms
65,632 KB
testcase_55 AC 228 ms
65,640 KB
testcase_56 AC 231 ms
65,640 KB
testcase_57 AC 231 ms
65,632 KB
testcase_58 AC 232 ms
65,640 KB
testcase_59 AC 231 ms
65,644 KB
testcase_60 AC 233 ms
65,632 KB
testcase_61 AC 230 ms
65,640 KB
testcase_62 AC 231 ms
65,640 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

const int Mod = 998244353;

typedef struct {
	int left, right, min, lc;
} 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].min = 0;
	v[k].lc = 0;
	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, int c)
{
	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].lc += c;
		v[k].min += c;
		for (j = k, i = j >> 1; i > 0; j = i, i >>= 1) v[i].min = (v[j].min < v[j^1].min)? v[j].min: v[j^1].min;
	} else {
		if (v[k].lc != 0) {
			j = k << 1;
			v[j].min += v[k].lc;
			v[j^1].min += v[k].lc;
			v[j].lc += v[k].lc;
			v[j^1].lc += v[k].lc;
			v[k].lc = 0;
		}
		update_segment(v, k << 1, l, r, c);
		update_segment(v, (k << 1) ^ 1, l, r, c);
	}
}

int get_min(lazy_seg_node v[], int k, int l, int r)
{
	int tmp[2];
	if (k == 1) update_segment(v, 1, l, r, 0);
	if (v[k].left > r || v[k].right < l) return 100;
	else if (v[k].left >= l && v[k].right <= r) return v[k].min;
	else {
		tmp[0] = get_min(v, k << 1, l, r);
		tmp[1] = get_min(v, (k << 1) ^ 1, l, r);
		return (tmp[0] < tmp[1])? tmp[0]: tmp[1];
	}
}

int BS_right(lazy_seg_node v[], int k, int l, int r)
{
	int j, tmp;
	if (v[k].lc != 0) {
		j = k << 1;
		v[j].min += v[k].lc;
		v[j^1].min += v[k].lc;
		v[j].lc += v[k].lc;
		v[j^1].lc += v[k].lc;
		v[k].lc = 0;
	}
	if (v[k].min > 0 || v[k].right < l || v[k].left > r) return l - 1;
	else if (v[k].left == v[k].right) return v[k].left;
	else {
		tmp = BS_right(v, (k << 1) ^ 1, l, r);
		if (tmp >= l) return tmp;
		else return BS_right(v, k << 1, l, r);
	}
}

typedef struct {
	int x, y;
} s_data;

typedef struct {
	s_data d[200001];
	int size;
} stack;

void push(stack* s, s_data z)
{
	s->d[++(s->size)] = z;
}

s_data pop(stack* s)
{
	return s->d[(s->size)--];
}

typedef struct {
	int key, id;
} data;

void merge_sort(int n, data x[])
{
	static data y[200001] = {};
	if (n <= 1) return;
	merge_sort(n / 2, &(x[0]));
	merge_sort((n + 1) / 2, &(x[n/2]));
	
	int i, p, q;
	for (i = 0, p = 0, q = n / 2; i < n; i++) {
		if (p >= n / 2) y[i] = x[q++];
		else if (q >= n) y[i] = x[p++];
		else y[i] = (x[p].key < x[q].key)? x[p++]: x[q++];
	}
	for (i = 0; i < n; i++) x[i] = y[i];
}

int main()
{
	int i, N, K, P[200001];
	scanf("%d %d", &N, &K);
	for (i = 1; i <= N; i++) scanf("%d", &(P[i]));
	
	s_data d;
	stack s_seg, s_min, s_max;
	s_seg.size = 0;
	s_min.size = 0;
	s_max.size = 0;
	d.x = 1; // node name
	d.y = 1; // left boundary
	push(&s_seg, d);
	d.x = P[1]; // value
	d.y = 1; // left boundary
	push(&s_min, d);
	push(&s_max, d);

	int j, k, u, n = N, par[400001], type[400001] = {}, left[400001];
	lazy_seg_node v[524288];
	init_node(v, 1, 1, N);
	for (u = 1; u <= N; u++) left[u] = u;
	for (i = 2; i <= N; i++) {
		j = i;
		while (s_min.size > 0) {
			d = pop(&s_min);
			if (d.x < P[i]) {
				push(&s_min, d);
				break;
			}
			update_segment(v, 1, d.y, j - 1, d.x - P[i]);
			j = d.y;
		}
		d.x = P[i];
		d.y = j;
		push(&s_min, d);
		
		j = i;
		while (s_max.size > 0) {
			d = pop(&s_max);
			if (d.x > P[i]) {
				push(&s_max, d);
				break;
			}
			update_segment(v, 1, d.y, j - 1, P[i] - d.x);
			j = d.y;
		}
		d.x = P[i];
		d.y = j;
		push(&s_max, d);

		update_segment(v, 1, 1, i - 1, -1);
		
		u = i;
		k = i;
		while (k > 1 && s_seg.size > 0) {
			j = BS_right(v, 1, 1, k - 1);
			if (j == 0) break;
			
			d = pop(&s_seg);
			if (j > d.y) {
				par[u] = d.x;
				u = par[u];
				k = d.y;
				continue;
			}
			
			par[u] = ++n;
			par[d.x] = n;
			left[n] = j;
			u = n;
			k = j;
			if (j == d.y) continue;
			
			type[n] = 1;
			while (d.y != j) {
				d = pop(&s_seg);
				par[d.x] = n;
			}
		}
		d.x = u;
		d.y = k;
		push(&s_seg, d);
	}
	
	int l[400001] = {}, w, r, deg[2][400001] = {}, q[400001], head, tail = 0, cur, prev;
	long long dp[400001][11] = {}, tmp[2][2][11];
	data *child[400001];
	for (u = 1; u <= n; u++) {
		if (par[u] != 0) deg[0][par[u]]++;
		else r = u;
	}
	for (u = 1; u <= n; u++) {
		if (deg[0][u] == 0) q[tail++] = u;
		else {
			child[u] = (data*)malloc(sizeof(data) * (deg[0][u] + 1));
			deg[1][u] = deg[0][u];
		}
	}
	for (u = 1; u <= n; u++) {
		if (u != r) {
			child[par[u]][l[par[u]]].key = left[u];
			child[par[u]][l[par[u]]++].id = u;
		}
	}
	for (head = 0; head < tail; head++) {
		u = q[head];
		if (deg[0][u] == 0) dp[u][1] = 1;
		else if (type[u] == 0) {
			merge_sort(deg[0][u], child[u]);
			for (j = 1, tmp[0][0][0] = 1, tmp[0][1][0] = 0; j <= K; j++) {
				tmp[0][0][j] = 0;
				tmp[0][1][j] = 0;
			}
			for (i = 0, cur = 1, prev = 0; i < deg[0][u]; i++, cur ^= 1, prev ^= 1) {
				w = child[u][i].id;
				for (j = 0; j <= K; j++) {
					tmp[cur][0][j] = 0;
					tmp[cur][1][j] = 0;
				}
				for (j = 0; j <= K; j++) {
					for (k = 1; j + k <= K; k++) tmp[cur][0][j+k] += tmp[prev][0][j] * dp[w][k] % Mod;
					tmp[cur][1][j] += tmp[prev][0][j] + tmp[prev][1][j];
					if (j < K) tmp[cur][0][j+1] += tmp[prev][1][j];
				}
				for (j = 0; j <= K; j++) {
					tmp[cur][0][j] %= Mod;
					tmp[cur][1][j] %= Mod;
				}
			}
			for (j = 1; j <= K; j++) dp[u][j] = tmp[prev][0][j];
		} else {
			for (i = 0, dp[u][0] = 1; i < deg[0][u]; i++) {
				w = child[u][i].id;
				for (j = K; j >= 0; j--) {
					for (k = j, dp[u][j] = 0; k > 0; k--) dp[u][j] += dp[u][j-k] * dp[w][k] % Mod;
					dp[u][j] %= Mod;
				}
			}
			dp[u][1]++;
		}
		
		if (u == r) continue;
		deg[1][par[u]]--;
		if (deg[1][par[u]] == 0) q[tail++] = par[u];
	}
	for (i = 1; i <= K; i++) printf("%lld\n", dp[r][i]);
	fflush(stdout);
	return 0;
}
0