結果
問題 | No.2340 Triple Tree Query (Easy) |
ユーザー | kotatsugame |
提出日時 | 2023-06-02 23:21:13 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 256 ms / 5,000 ms |
コード長 | 1,664 bytes |
コンパイル時間 | 1,445 ms |
コンパイル使用メモリ | 88,152 KB |
実行使用メモリ | 21,732 KB |
最終ジャッジ日時 | 2024-06-09 01:27:34 |
合計ジャッジ時間 | 10,936 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
5,760 KB |
testcase_01 | AC | 8 ms
6,144 KB |
testcase_02 | AC | 8 ms
5,888 KB |
testcase_03 | AC | 7 ms
6,016 KB |
testcase_04 | AC | 7 ms
6,016 KB |
testcase_05 | AC | 8 ms
6,016 KB |
testcase_06 | AC | 211 ms
14,168 KB |
testcase_07 | AC | 209 ms
14,036 KB |
testcase_08 | AC | 207 ms
14,040 KB |
testcase_09 | AC | 208 ms
14,168 KB |
testcase_10 | AC | 217 ms
14,048 KB |
testcase_11 | AC | 208 ms
14,168 KB |
testcase_12 | AC | 206 ms
14,168 KB |
testcase_13 | AC | 214 ms
14,296 KB |
testcase_14 | AC | 216 ms
14,168 KB |
testcase_15 | AC | 210 ms
14,176 KB |
testcase_16 | AC | 252 ms
21,732 KB |
testcase_17 | AC | 243 ms
19,228 KB |
testcase_18 | AC | 245 ms
18,676 KB |
testcase_19 | AC | 242 ms
21,148 KB |
testcase_20 | AC | 256 ms
18,644 KB |
testcase_21 | AC | 188 ms
14,456 KB |
testcase_22 | AC | 173 ms
14,452 KB |
testcase_23 | AC | 172 ms
14,456 KB |
testcase_24 | AC | 235 ms
14,344 KB |
testcase_25 | AC | 234 ms
14,188 KB |
testcase_26 | AC | 230 ms
14,212 KB |
testcase_27 | AC | 233 ms
14,224 KB |
testcase_28 | AC | 227 ms
14,092 KB |
testcase_29 | AC | 181 ms
14,444 KB |
testcase_30 | AC | 176 ms
14,436 KB |
testcase_31 | AC | 175 ms
14,320 KB |
testcase_32 | AC | 208 ms
14,468 KB |
testcase_33 | AC | 208 ms
14,220 KB |
testcase_34 | AC | 206 ms
14,348 KB |
testcase_35 | AC | 208 ms
14,344 KB |
testcase_36 | AC | 209 ms
14,340 KB |
ソースコード
#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); } } }