結果

問題 No.1496 不思議な数え上げ
ユーザー 👑 ygussanyygussany
提出日時 2021-05-01 00:01:16
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 257 ms / 3,000 ms
コード長 2,438 bytes
コンパイル時間 728 ms
コンパイル使用メモリ 31,900 KB
実行使用メモリ 11,928 KB
最終ジャッジ日時 2023-09-26 09:08:11
合計ジャッジ時間 12,176 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,384 KB
testcase_01 AC 1 ms
4,384 KB
testcase_02 AC 1 ms
4,384 KB
testcase_03 AC 163 ms
11,852 KB
testcase_04 AC 159 ms
11,916 KB
testcase_05 AC 161 ms
11,912 KB
testcase_06 AC 160 ms
11,884 KB
testcase_07 AC 160 ms
11,824 KB
testcase_08 AC 161 ms
11,776 KB
testcase_09 AC 158 ms
11,840 KB
testcase_10 AC 158 ms
11,840 KB
testcase_11 AC 159 ms
11,848 KB
testcase_12 AC 163 ms
11,884 KB
testcase_13 AC 152 ms
11,888 KB
testcase_14 AC 154 ms
11,884 KB
testcase_15 AC 151 ms
11,824 KB
testcase_16 AC 154 ms
11,836 KB
testcase_17 AC 257 ms
11,848 KB
testcase_18 AC 252 ms
11,824 KB
testcase_19 AC 249 ms
11,856 KB
testcase_20 AC 248 ms
11,856 KB
testcase_21 AC 249 ms
11,852 KB
testcase_22 AC 248 ms
11,848 KB
testcase_23 AC 248 ms
11,928 KB
testcase_24 AC 245 ms
11,780 KB
testcase_25 AC 248 ms
11,920 KB
testcase_26 AC 247 ms
11,876 KB
testcase_27 AC 230 ms
11,888 KB
testcase_28 AC 232 ms
11,928 KB
testcase_29 AC 232 ms
11,920 KB
testcase_30 AC 225 ms
11,752 KB
testcase_31 AC 230 ms
11,760 KB
testcase_32 AC 215 ms
11,596 KB
testcase_33 AC 107 ms
9,196 KB
testcase_34 AC 94 ms
8,732 KB
testcase_35 AC 3 ms
5,924 KB
testcase_36 AC 221 ms
11,676 KB
testcase_37 AC 115 ms
8,744 KB
testcase_38 AC 104 ms
9,000 KB
testcase_39 AC 153 ms
10,980 KB
testcase_40 AC 78 ms
8,524 KB
testcase_41 AC 154 ms
11,468 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

const int inf = 0;
int leaf[200001];

typedef struct {
	int left, right, max;
} seg_node;

void init_node(seg_node v[], int k, int l, int r)
{
	v[k].left = l;
	v[k].right = r;
	v[k].max = inf;
	if (l < r) {
		init_node(v, k << 1, l, (l + r) / 2);
		init_node(v, (k << 1) ^ 1, (l + r) / 2 + 1, r);
	} else leaf[l] = k;
}

void update_node(seg_node v[], int k, int x)
{
	int i, j = leaf[k];
	v[j].max = x;
	for (i = j >> 1; i > 0; j = i, i >>= 1) v[i].max = (v[j].max > v[j^1].max)? v[j].max: v[j^1].max;
}
 
int get_max(seg_node v[], int k, int l, int r)
{
	int tmp[2];
	if (v[k].right < l || v[k].left > r) return inf;
	else if (v[k].left >= l && v[k].right <= r) return v[k].max;
	else {
		tmp[0] = get_max(v, k << 1, l, r);
		tmp[1] = get_max(v, (k << 1) ^ 1, l, r);
		return (tmp[0] < tmp[1])? tmp[1]: tmp[0];
	}
}

int BS_left(seg_node v[], int k, int l, int r, int x)
{
	int tmp;
	if (v[k].max < x || v[k].right < l || v[k].left > r) return r + 1;
	else if (v[k].left == v[k].right) return v[k].left;
	else {
		tmp = BS_left(v, k << 1, l, r, x);
		if (tmp <= r) return tmp;
		else return BS_left(v, (k << 1) ^ 1, l, r, x);
	}
}

int BS_right(seg_node v[], int k, int l, int r, int x)
{
	int tmp;
	if (v[k].max < x || 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, x);
		if (tmp >= l) return tmp;
		else return BS_right(v, k << 1, l, r, x);
	}
}

int main()
{
	int i, N, P[200001], P_inv[200001];
	long long A, sum[200001], ans;
	scanf("%d", &N);
	for (i = 1; i <= N; i++) {
		scanf("%d", &(P[i]));
		P_inv[P[i]] = i;
	}
	for (i = 1, sum[0] = 0; i <= N; i++) sum[i] = sum[i-1] + P[i];
	
	int j, k, l, r, m, L, R;
	seg_node v[524288];
	init_node(v, 1, 1, N);
	for (i = 1; i <= N; i++) {
		scanf("%lld", &A);
		k = P_inv[i];
		L = BS_right(v, 1, 1, k - 1, 1) + 1;
		R = BS_left(v, 1, k + 1, N, 1) - 1;
		ans = 0;
		
		if (k - L < R - k) {
			for (j = k, r = R; j >= L; j--) {
				l = k - 1;
				while (l < r) {
					m = (l + r + 1) / 2;
					if (sum[m] - sum[j-1] <= A) l = m;
					else r = m - 1;
				}
				ans += l - k + 1;
			}
		} else {
			for (j = k, l = L; j <= R; j++) {
				r = k + 1;
				while (l < r) {
					m = (l + r) / 2;
					if (sum[j] - sum[m-1] <= A) r = m;
					else l = m + 1;
				}
				ans += k - l + 1;
			}
		}
		printf("%lld\n", ans);
		update_node(v, k, 1);
	}
	fflush(stdout);
	return 0;
}
0