結果

問題 No.1172 Add Recursive Sequence
コンテスト
ユーザー 👑 ygussany
提出日時 2020-08-15 00:17:04
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 322 ms / 4,000 ms
コード長 2,590 bytes
コンパイル時間 1,551 ms
コンパイル使用メモリ 33,024 KB
実行使用メモリ 6,656 KB
最終ジャッジ日時 2024-10-10 16:57:11
合計ジャッジ時間 3,045 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

#define Mod 1000000007

typedef struct {
	int key, id;
} data;

void merge_sort(data x[], int n)
{
	static data y[100001] = {};
	if (n <= 1) return;
	merge_sort(&(x[0]), n/2);
	merge_sort(&(x[n/2]), (n+1)/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];
}

typedef struct {
	data obj[100001];
	int size;
} min_heap;

void push(data x, min_heap* h)
{
	int i = h->size;
	data tmp;
	h->obj[(h->size)++] = x;
	while (1) {
		if (i == 0) break;
		else if (h->obj[i].key < h->obj[(i-1)/2].key) {
			tmp = h->obj[(i-1)/2];
			h->obj[(i-1)/2] = h->obj[i];
			h->obj[i] = tmp;
			i = (i - 1) / 2;
		} else break;
	}
}

data pop(min_heap* h)
{
	int i = 0, j;
	data output = h->obj[0], tmp;
	h->obj[0] = h->obj[--(h->size)];
	while (1) {
		if (i * 2 + 1 >= h->size) break;
		else if (i * 2 + 2 >= h->size) j = i * 2 + 1;
		else if (h->obj[i*2+1].key < h->obj[i*2+2].key) j = i * 2 + 1;
		else j = i * 2 + 2;
		if (h->obj[j].key < h->obj[i].key) {
			tmp = h->obj[j];
			h->obj[j] = h->obj[i];
			h->obj[i] = tmp;
			i = j;
		} else break;
	}
	return output;
}

int main()
{
	int i, j, k, K, N, M;
	long long a[100001], c[201];
	data d[100001];
	scanf("%d %d %d", &K, &N, &M);
	for (i = 0; i < K; i++) scanf("%lld", &(a[i]));
	for (i = 1; i <= K; i++) scanf("%lld", &(c[i]));
	for (i = 0; i < M; i++) scanf("%d %d", &(d[i].key), &(d[i].id));
	for (i = K; i < N; i++) {
		for (k = 1, a[i] = 0; k <= K; k++) a[i] = (a[i] + c[k] * a[i-k]) % Mod;
	}
	merge_sort(d, M);
	
	long long x[100001] = {}, tmp[100001] = {};
	data e;
	min_heap h[2];
	h[0].size = 0;
	h[1].size = 0;
	for (i = 0, j = 0; i < N; i++) {
		for (; j < M && d[j].key == i; j++) {
			if (d[j].id <= i + K) for (k = i; k < d[j].id; k++) x[k] += a[k-i];
			else {
				for (k = i; k < i + K; k++) x[k] += a[k-i];
				e.key = i + K;
				e.id = d[j].id;
				push(e, &(h[0]));
			}
		}
		if (i >= K) {
			while (h[0].size > 0 && h[0].obj[0].key == i) {
				e = pop(&(h[0]));
				for (k = i - 1; k >= i - K; k--) tmp[k] += a[k-(i-K)];
				e.key = e.id;
				e.id = i - K;
				push(e, &(h[1]));
			}
			while (h[1].size > 0 && h[1].obj[0].key == i) {
				e = pop(&(h[1]));
				for (k = i - 1; k >= i - K; k--) tmp[k] -= a[k-e.id];
			}
			for (k = i - 1; k >= i - K; k--) {
				tmp[k] = (tmp[k] % Mod + Mod) % Mod;
				tmp[i] = (tmp[i] + c[i-k] * tmp[k]) % Mod;
			}
			x[i] += tmp[i];
		}
		
		printf("%lld\n", x[i] % Mod);
	}
	fflush(stdout);
	return 0;
}
0