結果

問題 No.2273 一点乗除区間積
ユーザー 👑 potato167
提出日時 2023-04-14 22:10:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 871 ms / 5,000 ms
コード長 6,001 bytes
コンパイル時間 2,406 ms
コンパイル使用メモリ 214,580 KB
最終ジャッジ日時 2025-02-12 06:51:17
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
using namespace std;
using std::cout;
using std::cin;
using std::endl;
using ll=long long;
using ld=long double;
ll ILL=2167167167167167167;
const int INF=2100000000;
const int mod=998244353;
#define rep(i,a,b) for (ll i=a;i<b;i++)
#define all(p) p.begin(),p.end()
template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>;
template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";}
template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}
template<class T> T vec_min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;}
template<class T> T vec_max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;}
template<class T> T vec_sum(vector<T> &a){assert(!a.empty());T ans=a[0]-a[0];for(auto &x:a) ans+=x;return ans;}
namespace po167{
template <class T,T (*op)(T,T),T(*e)()>
struct segment_tree{
    int _n,size;
    std::vector<T> seg;
    int ceil_pow2(int a){
        int b=1;
        while(a>b){
            b<<=1;
        }
        return b;
    }
    void update(int k){seg[k]=op(seg[k*2],seg[k*2+1]);};
    segment_tree(int n) :_n(n){
        size=ceil_pow2(n);
        seg=std::vector<T>(size*2,e());
    }
    segment_tree(std::vector<T> &p) :_n((int) p.size()){
        size=ceil_pow2(_n);
        seg=std::vector<T>(size*2,e());
        for(int i=0;i<_n;i++) seg[i+size]=p[i];
        for(int i=size-1;i>0;i--) update(i);
    }
    void set(int ind,T val){
        assert(0<=ind&&ind<_n);
        ind+=size;
        seg[ind]=val;
        while(ind!=1){
            ind>>=1;
            update(ind);
        }
    }
    void addl(int ind,T val){
        set(ind,op(get(ind),val));
    }
    void addr(int ind,T val){
        set(ind,op(val,get(ind)));
    }
    T get(int ind){
        assert(0<=ind&&ind<_n);
        return seg[ind+size];
    }
    T query(int l,int r){
        assert(0<=l&&l<=r&&r<=_n);
        T l_val=e();
        T r_val=e();
        l+=size,r+=size;
        while(l<r){
            if(l&1) l_val=op(l_val,seg[l]),l+=1;
            if(r&1) r-=1,r_val=op(seg[r],r_val);
            r>>=1;
            l>>=1;
        }
        return op(l_val,r_val);
    }
    template <bool (*f)(T)> int max_right(int l) {
        return max_right(l, [](T x) { return f(x); });
    }
    template <class F> int max_right(int l, F f) {
        assert(0<=l&&l<=_n);
        assert(f(e()));
        if(f(query(l,_n))) return _n;
        T val=e();
        l+=size;
        while(true){
            while(l%2==0) l>>=1;
            if(!f(op(val,seg[l]))){
                while(l<size){
                    l*=2;
                    if(f(op(val,seg[l]))){
                        val=op(val,seg[l]);
                        l++;
                    }
                }
                return l-size;
            }
            val=op(val,seg[l]);
            l++;
        }
    }
    template <bool (*f)(T)> int min_left(int r) {
        return min_left(r, [](T x) { return f(x); });
    }
    template <class F> int min_left(int r, F f) {
        assert(0 <= r && r <= _n);
        assert(f(e()));
        if (r == 0) return 0;
        r += size;
        T sm = e();
        do {
            r--;
            while (r > 1 && (r % 2)) r >>= 1;
            if (!f(op(seg[r], sm))) {
                while (r < size) {
                    r = (2 * r + 1);
                    if (f(op(seg[r], sm))) {
                        sm = op(seg[r], sm);
                        r--;
                    }
                }
                return r + 1 - size;
            }
            sm = op(seg[r], sm);
        } while ((r & -r) != r);
        return 0;
    }
};
}
using po167::segment_tree;
// return val=p(N)
// a=p[0].first^p[0].second * ... *p[N-1].first^p[N-1].second
// for all i: p[i].first is prime number
// O(sqrt(val))
std::vector<std::pair<long long,long long>> Prime_factorization(long long val){
    assert(val>=1);
    if(val==1){
        return {};
    }
    int ind=0;
    std::vector<std::pair<long long,long long>> ans;
    for(long long i=2;i*i<=val;i++){
        if(val%i!=0) continue;
        ans.push_back({i,0});
        while(val%i==0){
            ans[ind].second++;
            val/=i;
        }
        ind++;
    }
    if(val!=1) ans.push_back({val,1});
    return ans;
}
int B;
using F= ll;
F op1(F a,F b){return (a*b)%B;}
F e1(){return 1;}
F op2(F a,F b){return a+b;}
F e2(){return 0;}
void solve();
// oddloop
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int t=1;
    //cin>>t;
    rep(i,0,t) solve();
}
ll jyo(ll x,ll y,ll z){
  ll H=y; //ここから
       ll a=1,b=(x%z+z)%z,c=1;
       while(H>0){
         a*=2;
         if(H%a!=0){
           H-=a/2;
           c*=b;
           c%=z;
         }
        b*=b;
         b%=z;
      } //ここまで
 return c;
}
void solve(){
    int N,Q;
    cin>>N>>B>>Q;
    auto p=Prime_factorization(B);
    vector<int> q;
    int L=p.size();
    for(auto x:p){
        int tmp=1;
        rep(i,0,x.second) tmp*=x.first;
        q.push_back(tmp);
    }
    auto f=[&](ll a)->pair<ll,vector<int>>{
        vector<int> res(L);
        if(a==0){
            return {0,res};
        }
        rep(i,0,L){
            while(a%p[i].first==0){
                res[i]++;
                a/=p[i].first;
            }
        }
        return {a,res};
    };
    segment_tree<F,op1,e1> seg1(N);
    segment_tree<F,op2,e2> em(N);
    vector<segment_tree<F,op2,e2>> seg2(L,em);
    rep(i,0,N){
        ll a;
        cin>>a;
        auto b=f(a);
        seg1.set(i,b.first);
        rep(j,0,L) seg2[j].set(i,b.second[j]);
    }
    rep(i,0,Q){
        ll j,m,l,r;
        cin>>j>>m>>l>>r;
        r++;
        bool ok=1;
        if(m==B){
            rep(k,0,L){
                if(seg2[k].get(j)<p[k].second) ok=0;
            }
        }else{
            ok=0;
        }
        if(ok){
            rep(k,0,L){
                seg2[k].addl(j,-p[k].second);
            }
        }else{
            auto b=f(m);
            rep(k,0,L){
                seg2[k].addl(j,b.second[k]);
            }
            seg1.addl(j,b.first);
        }
        auto ans=seg1.query(l,r);
        rep(k,0,L){
            ans=(ans*jyo(p[k].first,seg2.at(k).query(l,r),B))%B;
        }
        cout<<ans<<"\n";
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0