結果

問題 No.880 Yet Another Segment Tree Problem
コンテスト
ユーザー e869120
提出日時 2019-09-06 22:58:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4,599 ms / 5,000 ms
コード長 2,581 bytes
コンパイル時間 949 ms
コンパイル使用メモリ 75,256 KB
実行使用メモリ 6,656 KB
最終ジャッジ日時 2025-06-20 10:26:56
合計ジャッジ時間 35,940 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 38
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:91:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   91 |         scanf("%d%d", &N, &Q);
      |         ~~~~~^~~~~~~~~~~~~~~~
main.cpp:92:43: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   92 |         for (int i = 1; i <= N; i++) scanf("%d", &A[i]);
      |                                      ~~~~~^~~~~~~~~~~~~
main.cpp:94:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   94 |                 scanf("%d%d%d", &T[i], &L[i], &R[i]); R[i]++;
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:95:50: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   95 |                 if (T[i] == 1 || T[i] == 2) scanf("%d", &X[i]);
      |                                             ~~~~~^~~~~~~~~~~~~

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#pragma warning (disable: 4996)

const int BACKET = 500;
int N, Q, A[1 << 18], T[1 << 18], L[1 << 18], R[1 << 18], X[1 << 18];
long long ans[1 << 18];
bool used[1 << 15];
int val[1 << 15], v_max[1 << 15]; long long v_sum[1 << 15];

int gcd(int a, int b) {
	if (b == 0) return a;
	return gcd(b, a % b);
}

void ranged(int l, int r) {
	// 分割
	vector<int> C;
	for (int i = l; i < r; i++) { C.push_back(L[i]); C.push_back(R[i]); }
	sort(C.begin(), C.end());
	C.erase(unique(C.begin(), C.end()), C.end());

	// 初期化
	for (int i = 0; i < C.size(); i++) { used[i] = false; val[i] = 0; v_max[i] = 0; v_sum[i] = 0; }
	for (int i = 0; i < C.size() - 1; i++) {
		for (int j = C[i]; j < C[i + 1]; j++) {
			v_max[i] = max(v_max[i], A[j]);
			v_sum[i] += A[j];
		}
	}

	// クエリに答える
	for (int i = l; i < r; i++) {
		int pos1 = lower_bound(C.begin(), C.end(), L[i]) - C.begin();
		int pos2 = lower_bound(C.begin(), C.end(), R[i]) - C.begin();

		// 置換
		if (T[i] == 1) {
			for (int j = pos1; j < pos2; j++) {
				used[j] = true;
				val[j] = X[i];
			}
		}
		// gcd
		if (T[i] == 2) {
			for (int j = pos1; j < pos2; j++) {
				if (used[j] == true) {
					val[j] = gcd(val[j], X[i]);
				}
				if (used[j] == false) {
					int e = gcd(val[j], X[i]);
					if (e == val[j]) continue;

					val[j] = e; v_max[j] = 0; v_sum[j] = 0;
					for (int k = C[j]; k < C[j + 1]; k++) {
						int ee = gcd(e, A[k]);
						v_max[j] = max(v_max[j], ee);
						v_sum[j] += ee;
					}
				}
			}
		}
		// max を求める
		if (T[i] == 3) {
			for (int j = pos1; j < pos2; j++) {
				if (used[j] == true) ans[i] = max((int)ans[i], val[j]);
				else ans[i] = max((int)ans[i], v_max[j]);
			}
		}
		// 総和を求める
		if (T[i] == 4) {
			for (int j = pos1; j < pos2; j++) {
				if (used[j] == true) ans[i] += 1LL * (C[j + 1] - C[j]) * val[j];
				else ans[i] += v_sum[j];
			}
		}
	}

	// 更新
	for (int i = 0; i < C.size() - 1; i++) {
		for (int j = C[i]; j < C[i + 1]; j++) {
			if (used[i] == true) A[j] = val[i];
			else A[j] = gcd(val[i], A[j]);
		}
	}
}

int main() {
	scanf("%d%d", &N, &Q);
	for (int i = 1; i <= N; i++) scanf("%d", &A[i]);
	for (int i = 1; i <= Q; i++) {
		scanf("%d%d%d", &T[i], &L[i], &R[i]); R[i]++;
		if (T[i] == 1 || T[i] == 2) scanf("%d", &X[i]);
	}
	for (int i = 1; i <= Q; i += BACKET) {
		int cl = i, cr = i + BACKET; cr = min(cr, (int)Q + 1);
		ranged(cl, cr);
	}
	for (int i = 1; i <= Q; i++) {
		if (T[i] == 3 || T[i] == 4) printf("%lld\n", ans[i]);
	}
	return 0;
}
0