ABC417——F - Random Gathering

思路讲解

就是简单的东西。

FOR(_,1,M){
    ll l,r;
    cin>>l>>r;
    ll ret=tr.query(l, r);
    tr.modify(l, r, 0);
    ll res=ret*binpow(r-l+1, mod-2)%mod;
    tr.add(l, r, res);
}
FOR(i,1,tr.sz-1){
    tr.pushdown(i);
}
FOR(i,1,N){
    ll res=tr.tr[i+tr.sz];
    if(res<0) res+=mod;
    cout<<res<<" ";
}

AC代码

源代码

// by Gospel_rock
#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define PB push_back
#define EB emplace_back
#define EM emplace
#define FI first
#define SE second
#define pct __builtin_popcountll
#define ctz __builtin_ctzll
#define SZ(a) ((long long) a.size())
#define FOR(i, a, b) for (long long i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (long long i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define cend cerr<<"\n-----------\n"
#define fsp(x) fixed<<setprecision(x)
#define minn(vec) *min_element(vec.begin(),vec.end())
#define maxx(vec) *max_element(vec.begin(),vec.end())
using namespace std;
#ifdef LOCAL
template<typename T> void _print(T x) { cerr << x; }
// 可以打印多层数组，但是格式不是很好
template<typename T> void _print(vector<T> v) { cerr << "[ "; for (T i : v) _print(i), cerr << " "; cerr << "]"; }
template<typename T> void _print(set<T> s) { cerr << "{ "; for (T i : s) _print(i), cerr << " "; cerr << "}"; }
template<typename T, typename U> void _print(pair<T, U> p) { cerr << "{"; _print(p.first); cerr << ", "; _print(p.second); cerr << "}"; }
template<typename T, typename U> void _print(map<T, U> m) { cerr << "{ "; for (auto &p : m) _print(p), cerr << " "; cerr << "}"; }
// 打印两层数组，格式好一些
template<typename T> void _print(vector<vector<T>> v){cerr<<"{\n";for (auto i : v) _print(i), cerr << "\n";cerr<<" }";}
#endif

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;using vll=vector<ll>;using vb=vector<bool>;
using tll = tuple<ll,ll,ll>;using vii=vector<int>;
using vpll = vector<pll>;using vtll = vector<tll>;
using vdb = vector<DB>;using vc=vector<char>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
using CD=complex<double>;
static constexpr ll MAXN=(ll)1e6+10,INF=(ll)1e17+3; // 1e17+3是素数 1e18+9是素数
static constexpr ll mod=998244353;
static constexpr double eps=1e-8;const double pi=acos(-1.0);
ll N,M,Q,X,K,T,lT;

inline void __();
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    
    // cin>>lT;
    // while(lT--)
        __();
    return 0;
}
/*

*/
// 参考zkw的PPT 以及 https://www.luogu.com.cn/record/207546670
// 参考题解 https://www.luogu.com.cn/article/w3gfeebh
// AC https://atcoder.jp/contests/abc417/submissions/68719700 线段树区间加减，区间修改，区间和查询
constexpr ll preset=0,initv=-INF;
// 依赖常量mod   线段树区间加减，区间修改，区间和查询
struct SEG{
    int n,sz=1,dep;
    static ll op(const ll &a,const ll &b){
        return (a+b)%mod;
    }
    vector<ll> tr;vector<ll> tag;vector<ll> tag2;
    SEG(int n){     // 传入的时候可以传大一点
        this->n=n+2;
        dep=__lg(this->n)+1;
        sz=1<<dep;
        tr.resize((sz<<1)+5,preset);
        tag.resize((sz<<1)+5,preset);
        tag2.resize((sz<<1)+5,initv);
    }
    int len(int o){return 1<<(dep-__lg(o));};
    void pushup(int o){tr[o]=op(tr[o<<1],tr[o<<1|1]);}
    void settag(int o,ll x){
        tr[o]=op(tr[o],x*len(o));
        tag[o]=op(tag[o],x);
    }
    void settag2(int o,ll x){
        tr[o]=op(len(o)*x,0);
        tag[o]=0;
        tag2[o]=x;
    }
    void pushdown(int o){
        int l=o<<1,r=l|1;   // 左儿子，右儿子
        if(tag2[o]!=initv){
            settag2(l, tag2[o]);
            settag2(r, tag2[o]);
            tag2[o]=initv;
        }
        if(tag[o]!=0){
            settag(l, tag[o]);
            settag(r, tag[o]);
            tag[o]=0;
        }
    }
    void assign(int i,ll x){
        tr[i+sz]=x;
    }
    void build(){
        for(int o=sz-1;o>=1;--o){
            pushup(o);
        }
    }
    void push(int l,int r){
        //        在j层，l，r变为了同一点
        int i=dep,j=__lg(l^r);
        while (i>j) {   // 就是为了避免一下重复下推
            pushdown(l>>i--);
        }
        while(i){
            pushdown(l>>i),pushdown(r>>i--);
        }
    }
    ll query(int l,int r){
        ll res=preset;
        // 变为开区间，并加上sz
        l=l-1+sz,r=r+1+sz;
        push(l,r);
        for(;l^r^1;l>>=1,r>>=1){
            // l^r^1 表示 只要 l 与 r 还不是同一个父亲的两兄弟，就继续上跳。
            if(~l&1){   
                // l是偶数，说明l是父节点的左儿子，开区间，将其右儿子加入答案
                res=op(res,tr[l^1]);
            }
            if(r&1){
                res=op(tr[r^1],res);
            }
        }
        return res;
    }
    void add(int l,int r,ll x){
        l=l-1+sz,r=r+1+sz;
        push(l, r);
        for(;l^r^1;pushup(l>>=1),pushup(r>>=1)){
            if(~l&1){
                settag(l^1, x);
            }
            if(r&1){
                settag(r^1, x);
            }
        }
        for (l >>= 1; l; l >>= 1) pushup(l);
    }
    void modify(int l,int r,ll x){
        l=l-1+sz,r=r+1+sz;
        push(l, r);
        for(;l^r^1;pushup(l>>=1),pushup(r>>=1)){
            if(~l&1){
                settag2(l^1, x);
            }
            if(r&1){
                settag2(r^1, x);
            }
        }
        for (l >>= 1; l; l >>= 1) pushup(l);
    }
};
ll binpow(ll a,ll k,const ll &p=mod){
    ll res=1;
    a%=p;
    while(k>0){
        if((k&1)==1) res=a*res%p;
        a=a*a%p;
        k>>=1;
    }
    return res;
}
inline void __(){
    cin>>N>>M;
    SEG tr(N);
    FOR(i,1,N){
        ll a;cin>>a;
        tr.assign(i, a);
    }
    tr.build();
    FOR(_,1,M){
        ll l,r;
        cin>>l>>r;
        ll ret=tr.query(l, r);
        tr.modify(l, r, 0);
        ll res=ret*binpow(r-l+1, mod-2)%mod;
        tr.add(l, r, res);
    }
    FOR(i,1,tr.sz-1){
        tr.pushdown(i);
    }
    FOR(i,1,N){
        ll res=tr.tr[i+tr.sz];
        if(res<0) res+=mod;
        cout<<res<<" ";
    }
}
/*
AC
https://atcoder.jp/contests/abc417/submissions/68719700
*/

心路历程（WA，TLE，MLE……）