#1041276 (C++17) No.3524 二進範囲更新範囲和取得

• 問題: No.3524 二進範囲更新範囲和取得
• ユーザー: 👑 binap
• 提出日時: 2025-02-07 18:34:22
• 言語: C++17 (gcc 15.2.0 + boost 1.89.0)
• 結果: MLE
• 実行時間: -
• コード長: 3,953 bytes
• コンパイル時間: 5,085 ms
• コンパイル使用メモリ: 286,968 KB
• 実行使用メモリ: 1,076,012 KB
• 平均クエリ数: 267.69
• 最終ジャッジ日時: 2026-05-01 20:58:06
• 合計ジャッジ時間: 61,747 ms
• ジャッジサーバーID （参考情報）: judge3_0 / judge1_0

ファイルパターン	結果
other	AC * 18 MLE * 27

ソースコード

#include<bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef vector<int> vi;
typedef vector<long long> vl;
typedef vector<vector<int>> vvi;
typedef vector<vector<long long>> vvl;
typedef long double ld;
typedef pair<int, int> P;

template <int m> ostream& operator<<(ostream& os, const static_modint<m>& a) {os << a.val(); return os;}
template <int m> ostream& operator<<(ostream& os, const dynamic_modint<m>& a) {os << a.val(); return os;}
template <int m> istream& operator>>(istream& is, static_modint<m>& a) {long long x; is >> x; a = x; return is;}
template <int m> istream& operator>>(istream& is, dynamic_modint<m>& a) {long long x; is >> x; a = x; return is;}
template<typename T> istream& operator>>(istream& is, vector<T>& v){int n = v.size(); assert(n > 0); rep(i, n) is >> v[i]; return is;}
template<typename U, typename T> ostream& operator<<(ostream& os, const pair<U, T>& p){os << p.first << ' ' << p.second; return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<T>& v){int n = v.size(); rep(i, n) os << v[i] << (i == n - 1 ? "\n" : " "); return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<vector<T>>& v){int n = v.size(); rep(i, n) os << v[i] << (i == n - 1 ? "\n" : ""); return os;}
template<typename T> ostream& operator<<(ostream& os, const set<T>& se){for(T x : se) os << x << " "; os << "\n"; return os;}
template<typename T> ostream& operator<<(ostream& os, const unordered_set<T>& se){for(T x : se) os << x << " "; os << "\n"; return os;}
template<typename S, auto op, auto e> ostream& operator<<(ostream& os, const atcoder::segtree<S, op, e>& seg){int n = seg.max_right(0, [](S){return true;}); rep(i, n) os << seg.get(i) << (i == n - 1 ? "\n" : " "); return os;}
template<typename S, auto op, auto e, typename F, auto mapping, auto composition, auto id> ostream& operator<<(ostream& os, const atcoder::lazy_segtree<S, op, e, F, mapping, composition, id>& seg){int n = seg.max_right(0, [](S){return true;}); rep(i, n) os << seg.get(i) << (i == n - 1 ? "\n" : " "); return os;}

template<typename T> void chmin(T& a, T b){a = min(a, b);}
template<typename T> void chmax(T& a, T b){a = max(a, b);}

// https://ei1333.github.io/luzhiled/snippets/math/mod-pow.html
template< typename T >
T mod_pow(T x, T n, const T &p) {
  T ret = 1;
  while(n > 0) {
    if(n & 1) (ret *= x) %= p;
    (x *= x) %= p;
    n >>= 1;
  }
  return ret;
}

using T = uint8_t;

int B;
using S = vector<T>;
S zero;
S op(S a, S b){
	rep(i, B){
		a[i] += b[i]; 
		if(a[i] >= B) a[i] -= B;
	}
	return a;
}
S e(){
	return zero;
}
using F = vector<T>;
S mapping(F f, S s){
	S res(B);
	rep(i, B){
		res[f[i]] += s[i];
		if(res[f[i]] >= B) res[f[i]] -= B;
	}
	return res;
}
F composite(F f, F g){
	rep(i, B) g[i] = f[g[i]];
	return g;
}
F identical;
F id(){
	return identical;
}

int main(){
	int N, Q;
	cin >> N >> B >> Q;
	
	zero.resize(B);
	identical.resize(B);
	rep(i, B) identical[i] = i;
	
	vector<int> in(N + 1);
	vector<int> out(N + 1);
	int t = 0;
	
	auto dfs = [&](int from, auto dfs) -> void{
		if(from > N) return;
		in[from] = t; t++;
		dfs(from * 2, dfs);
		dfs(from * 2 + 1, dfs);
		out[from] = t;
	};
	dfs(1, dfs);
	
	vector<int> a(N);
	
	for(int x = 1; x <= N; x++){
		int idx = in[x];
		a[idx] = x % B;
	}
	
	vector<S> _init;
	{
		rep(idx, N){
			S tmp(B);
			tmp[a[idx]] = 1;
			_init.push_back(tmp);
		}
	}
	lazy_segtree<S, op, e, F, mapping, composite, id> seg(_init);
	
	rep(_, Q){
		int x;
		long long d;
		cin >> x >> d;
		
		vector<T> transfer_sub(B);
		rep(j, B) transfer_sub[j] = (mod_pow<long long>(j, d, B) + 1) % B;
		
		int ans = 0;
		
		seg.apply(in[x], out[x], transfer_sub);
		auto res = seg.prod(in[x], out[x]);
		rep(i, B){
			ans += (int)i * res[i];
			ans %= B;
		}
		
		/*
		for(auto x : res) cout << x << ' ';
		cout << "\n";
		*/
		cout << ans << "\n";
	}
	
	return 0;
}