#979186 (C++17) No.880 Yet Another Segment Tree Problem

提出ソース

結果

問題	No.880 Yet Another Segment Tree Problem
ユーザー	t9unkubj
提出日時	2024-05-05 23:50:41
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	8,382 bytes
コンパイル時間	4,192 ms
コンパイル使用メモリ	241,072 KB
実行使用メモリ	15,104 KB
最終ジャッジ日時	2024-05-05 23:51:06
合計ジャッジ時間	20,224 ms
ジャッジサーバーID （参考情報）	judge2 / judge5

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	WA	-
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	AC	511 ms 14,976 KB
testcase_34	AC	519 ms 14,976 KB
testcase_35	AC	497 ms 14,976 KB
testcase_36	AC	517 ms 14,976 KB
testcase_37	AC	509 ms 14,976 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#pragma GCC optimize("O3")
#ifdef t9unkubj
#include "debug.cpp"
#else
#define dbg(...) 199958
#endif

using namespace std;
#include<bits/stdc++.h>
using uint=unsigned;
using ll=long long;
using ull=unsigned long long;
using ld=long double;
using pii=pair<int,int>;
using pll=pair<ll,ll>;
template<class T>using vc=vector<T>;
template<class T>using vvc=vc<vc<T>>;
template<class T>using vvvc=vvc<vc<T>>;
using vi=vc<int>;
using vvi=vc<vi>;
using vvvi=vc<vvi>;
using vl=vc<ll>;
using vvl=vc<vl>;
using vvvl=vc<vvl>;
template<class T>using smpq=priority_queue<T,vector<T>,greater<T>>;
template<class T>using bipq=priority_queue<T>;
#define rep(i,n) for(ll i=0;i<(ll)n;i++)
#define REP(i,j,n) for(ll i=j;i<(ll)n;i++)
#define DREP(i,n,m) for(ll i=n;i>=m;i--)
#define drep(i,n) for(ll i=n-1;i>=0;i--)
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define mp make_pair
#define mt make_tuple
#define pb push_back
#define eb emplace_back
#define fi first
#define se second
#define is insert
#define bg begin()
#define ed end()
void scan(int&a) { cin >> a; }
void scan(ll&a) { cin >> a; }
void scan(string&a) { cin >> a; }
void scan(char&a) { cin >> a; }
void scan(uint&a) { cin >> a; }
void scan(ull&a) { cin >> a; }
void scan(bool&a) { cin >> a; }
template<class T> void scan(vector<T>&a) { for(auto&x:a) scan(x); }
void read() {}
template<class Head, class... Tail> void read(Head&head, Tail&... tail) { scan(head); read(tail...); }
#define INT(...) int __VA_ARGS__; read(__VA_ARGS__);
#define LL(...) ll __VA_ARGS__; read(__VA_ARGS__);
#define ULL(...) ull __VA_ARGS__; read(__VA_ARGS__);
#define STR(...) string __VA_ARGS__; read(__VA_ARGS__);
#define CHR(...) char __VA_ARGS__; read(__VA_ARGS__);
#define DBL(...) double __VA_ARGS__; read(__VA_ARGS__);
#define LD(...) ld __VA_ARGS__; read(__VA_ARGS__);
#define VC(type, name, ...) vector<type> name(__VA_ARGS__); read(name);
#define VVC(type, name, size, ...) vector<vector<type>> name(size, vector<type>(__VA_ARGS__)); read(name);
void print(int a) { cout << a; }
void print(ll a) { cout << a; }
void print(string&a) { cout << a; }
void print(char&a) { cout << a; }
void print(uint&a) { cout << a; }
void print(bool&a) { cout << a; }
void print(ull&a) { cout << a; }
template<class T> void print(vector<T>&a) { for(int i=0;i<(int)a.size();i++){if(i)cout<<" ";print(a[i]);}cout<<endl;}
int PRT() { cout << 'n'; return 0; }
template<class T> int PRT(T a) { print(a); cout <<endl; return 0; }
template<class Head, class... Tail> int PRT(Head head, Tail ... tail) { print(head); cout << " "; PRT(tail...); return 0; }
template<class T,class F>
bool chmin(T &x, F y){
    if(x>y){
        x=y;
        return true;
    }
    return false;
}
template<class T, class F>
bool chmax(T &x, F y){
    if(x<y){
        x=y;
        return true;
    }
    return false;
}
void YesNo(bool b){
    cout<<(b?"Yes":"No")<<endl;
}
void Yes(){
    cout<<"Yes"<<endl;
}
void No(){
    cout<<"No"<<endl;
}
template<class T>
int popcount(T n){
    return __builtin_popcount(n);
}
template<class T>
T sum(vc<T>&a){
    return accumulate(all(a),T(0));
}
template<class T>
T max(vc<T>&a){
    return *max_element(all(a));
}
template<class T>
T min(vc<T>&a){
    return *min_element(all(a));
}
template<class T>
void unique(vc<T>&a){
    a.erase(unique(all(a)),a.end());
}
vvi readgraph(int n,int m,int off = -1){
    vvi g(n);
    rep(i, m){
        int u,v;
        cin>>u>>v;
        u+=off,v+=off;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    return g;
}
vvi readtree(int n,int off=-1){
    return readgraph(n,n-1,off);
}
template<class T>
vc<T> presum(vc<T> &a){
    vc<T> ret(a.size()+1);
    rep(i,a.size())ret[i+1]=ret[i]+a[i];
    return ret;
}
template<class T, class F>
vc<T> &operator+=(vc<T> &a,F b){
    for (auto&v:a)v += b;
    return a;
}
template<class T, class F>
vc<T> &operator-=(vc<T>&a,F b){
    for (auto&v:a)v-=b;
    return a;
}
template<class T, class F>
vc<T> &operator*=(vc<T>&a,F b){
    for (auto&v:a)v*=b;
    return a;
}
double pass_time=0;
template<class T,class F>
struct lazy_segtree{
    int n;
    vector<T>data;
    vector<F>lazy;
    using TT=function<T(T,T)>;
    using TF=function<T(T,F)>;
    using FF=function<F(F,F)>;
    TT op;
    T e;
    TF mapping;
    FF merge;
    F id;
    int log;
    lazy_segtree(int n1,TT op,T e,TF mapping,FF merge,F id):n(n1),op(op),e(e),mapping(mapping),merge(merge),id(id){
        int n_=1;
        log=0;
        while(n_<=n)n_*=2,log++;
        data=vector<T>(n_*2,e);
        lazy=vector<F>(n_,id);
        n=n_;
        for(int i=n_-1;i>0;i--)update(i);
    }
    void update(int k){
        data[k]=op(data[k*2],data[k*2+1]);
    }
    void apply(int k,F x){//
        data[k]=mapping(data[k],x);
        if(k<n){
            lazy[k]=merge(lazy[k],x);
            if(lazy[k]!=id)
            //segment tree beats! 
            if(data[k].fail)push(k),update(k);
        }
    }
    void push(int k){//上から遅延素を伝播させる
        apply(k*2,lazy[k]);
        apply(k*2+1,lazy[k]);
        lazy[k]=id;
    }
    void set(int p,T x){
        p+=n;
        for(int i=log;i>=1;i--){
            push(p>>i);
        }
        data[p]=x;
        for(int i=1;i<=log;i++){
            update(p>>i);
        }
    }
    void apply(int l,int r,F x){
        if(l==r)return;
        l+=n,r+=n;
        for(int i=log;i>=1;i--){
            if(((l>>i)<<i)!=l)push(l>>i);
            if(((r>>i)<<i)!=r)push((r-1)>>i);
        } 
        int l2{l},r2{r};
        while(l<r){
            if(l&1)apply(l++,x);
            if(r&1)apply(--r,x);
            l>>=1,r>>=1;
        }
        l=l2,r=r2;
        for(int i=1;i<=log;i++){
            if(((l>>i)<<i)!=l)update(l>>i);
            if(((r>>i)<<i)!=r)update((r-1)>>i);
        }
    }
    T prod(int l,int r){
        l+=n,r+=n;
        for(int i=log;i>=1;i--){
            if(((l>>i)<<i)!=l)push(l>>i);
            if(((r>>i)<<i)!=r)push((r-1)>>i);
        } 
        T sml=e,smr=e;
        while(l<r){
            if(l&1)sml=op(sml,data[l++]);
            if(r&1)smr=op(data[--r],smr);
            l>>=1,r>>=1;
        }
        return op(sml,smr);
    }
    T get(int p){
        p+=n;
        for(int i=log;i>=1;i--)push(p>>i);
        return data[p];
    }
};
ll INF=1ll<<30;
struct S{
    ll mx,lcm,sz;
    ll sum;
    bool same;
    bool fail;  
    S():mx(0),lcm(1),sz(0),sum(0),fail(0),same(1){}
    S(ll x,ll siz):mx(x),lcm(x),sz(siz),sum(x*siz),fail(0),same(1){}
};
struct F{
    int gcd,update;
    F():gcd(0),update(0){}
};
bool operator!=(F a,F b){
    return a.gcd!=b.gcd||a.update!=b.update;
}
F gcd_(int x){
    F res;
    res.gcd=x;
    res.update=0;
    return res;
}
F update_(int x){
    F res;
    res.gcd=0;
    res.update=x;
    return res;
}
void solve(){
    auto op=[](S a,S b){
        if(a.sz==0)return b;
        if(b.sz==0)return a;
        S c(0,0);
        c.mx=max(a.mx,b.mx);
        c.lcm=min<ll>(INF,a.lcm*b.lcm/gcd(a.lcm,b.lcm));
        c.sum=a.sum+b.sum;
        c.sz=a.sz+b.sz;
        c.same=(a.mx==b.mx)&&(a.same)&&(b.same);
        return c;
    };
    auto mapping=[](S a,F b){
        if(a.fail)return a;
        if(b.update)return S(b.update,a.sz);
        if(b.gcd){
            if(a.sz==1||a.same){
                a=S(gcd(a.mx,b.gcd),1);
            }else if(a.lcm==INF||b.gcd%a.lcm){
                a.fail=1;
            }
        }
        return a;
    };
    auto merge=[](F a,F b){
        if(b.update)return b;
        if(a.update){
            return update_(gcd(a.update,b.gcd));
        }
        return gcd_(gcd(a.gcd,b.gcd));
    };
    INT(n,q);
    VC(int,a,n);
    lazy_segtree<S,F>seg(n,op,S(),mapping,merge,F());
    rep(i,n)seg.set(i,S(a[i],1));
    while(q--){
        INT(t,l,r);
        --l,--r;
        if(t==1){
            INT(x);
            seg.apply(l,r+1,update_(x));
        }
        if(t==2){
            INT(x);
            seg.apply(l,r+1,gcd_(x));
        }
        if(t==3){
            PRT(seg.prod(l,r+1).mx);
        }
        if(t==4){
            PRT(seg.prod(l,r+1).sum);
        }
    }
}
signed main(){
    #ifdef t9unkubj
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    #endif
    cin.tie(0)->sync_with_stdio(0);
    pass_time=clock();
    int t=1;
    //cin>>t;
    while(t--)solve();
    pass_time=clock()-pass_time;
    dbg(pass_time/CLOCKS_PER_SEC);
}

yukicoder

結果

テストケース

ソースコード