結果
問題 | No.2340 Triple Tree Query (Easy) |
ユーザー |
|
提出日時 | 2023-06-02 23:21:13 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 222 ms / 5,000 ms |
コード長 | 1,664 bytes |
コンパイル時間 | 1,116 ms |
コンパイル使用メモリ | 88,280 KB |
実行使用メモリ | 21,604 KB |
最終ジャッジ日時 | 2024-12-28 22:33:57 |
合計ジャッジ時間 | 10,250 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 36 |
ソースコード
#include<iostream>#include<vector>#include<cassert>#include<atcoder/modint>#include<atcoder/lazysegtree>using namespace std;using mint=atcoder::modint998244353;mint op(mint a,mint b){return a;}mint e(){return mint::raw(0);}using F=pair<mint,mint>;mint mp(F f,mint x){return f.first*x+f.second;}F cmp(F f,F g){return make_pair(f.first*g.first,f.first*g.second+f.second);}F id(){return make_pair(mint::raw(1),mint::raw(0));}int N,Q;vector<int>G[100000];vector<int>V;int inv[100000],pr[100000];int cL[100000],cR[100000];int vL[100000],vR[100000];void dfs(int u,int p){pr[u]=p;cL[u]=vL[u]=V.size();for(int v:G[u])if(v!=p){inv[v]=V.size();V.push_back(v);}cR[u]=V.size();for(int v:G[u])if(v!=p)dfs(v,u);vR[u]=V.size();}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin>>N>>Q;for(int i=1;i<N;i++){int a,b;cin>>a>>b;a--,b--;G[a].push_back(b);G[b].push_back(a);}inv[0]=0;V.push_back(0);dfs(0,-1);vector<mint>init(N);for(int i=0;i<N;i++){int x;cin>>x;init[inv[i]]=mint::raw(x);}atcoder::lazy_segtree<mint,op,e,F,mp,cmp,id>seg(init);for(int i=0;i<Q;i++){int op;cin>>op;if(op==1){int v;cin>>v;v--;cout<<seg.get(inv[v]).val()<<"\n";}else if(op==2){int v,k,c,d;cin>>v>>k>>c>>d;v--;F f=make_pair(mint::raw(c),mint::raw(d));seg.set(inv[v],mp(f,seg.get(inv[v])));if(pr[v]!=-1)seg.set(inv[pr[v]],mp(f,seg.get(inv[pr[v]])));seg.apply(cL[v],cR[v],f);}else{int v,c,d;cin>>v>>c>>d;v--;F f=make_pair(mint::raw(c),mint::raw(d));seg.set(inv[v],mp(f,seg.get(inv[v])));seg.apply(vL[v],vR[v],f);}}}