oj.uz

Submission #637964

# Submission time Handle Problem Language Result Execution time Memory
637964 2022-09-03T18:30:04 Z victor_gao Sprinkler (JOI22_sprinkler) C++17
3 / 100
 997 ms 1048576 KB 
//#pragma GCC optimize("Ofast,unroll-loops,O3")
//#pragma GCC optimize("avx,avx2,sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,fma,tune=native")
#include<bits/stdc++.h>
//#include<bits/extc++.h>
//#pragma pack(1)
#define fast ios::sync_with_stdio(0); cin.tie(0);
#define int long long
#define pii pair<int,int>
#define x first
#define y second
#define N 200015
using namespace std;
//using namespace __gnu_pbds;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
//typedef tree<int, null_type,less_equal<int>, rb_tree_tag,tree_order_statistics_node_update> order_multiset;
//typedef tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update> order_set;
int n,L,loc[N],sz[N],fa[N],h[N];
vector<int>g[N],son[N];
struct BIT_2D{
    vector<int>bit[45];
    int n=0,m=43;
    void init(int _n){
        n=_n;
        for (int i=0;i<=m;i++)
            bit[i].resize(n+5,1);
    }
    void modify(int x,int y,int c){
        x=m-x;
       // cout<<"modify : "<<x<<' '<<y<<" "<<c<<'\n';
        while (x<=m){
            int ty=y;
            while (ty<=n){
                bit[x][ty]=bit[x][ty]*c%L;
                ty+=ty&(-ty);
            }
            x+=x&(-x);
        }
    }
    int query(int x,int y){
        int ans=1;
        x=m-x;
      //  cout<<"query : "<<x<<' '<<y<<'\n';
        while (x>0){
            int ty=y;
            while (ty>0){
                ans=ans*bit[x][ty]%L;
                ty-=ty&(-ty);
            }
            x-=x&(-x);
        }
        return ans;
    }
}cntl[N],cntr[N];
void dfs(int p,int lp){
    fa[p]=lp;
    for (auto i:g[p]){
        if (i!=lp){
            dfs(i,p);
            son[p].push_back(i);
            loc[i]=son[p].size();
        }
    }
    sz[p]=son[p].size();
}
signed main(){
    fast
    cin>>n>>L;
    for (int i=1;i<n;i++){
        int a,b; cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    for (int i=1;i<=n;i++)
        cin>>h[i];
    dfs(1,0);
    loc[1]=1; sz[0]=1;
    for (int i=0;i<=n;i++){
        sz[i]++;
        cntl[i].init(sz[i]);
        cntr[i].init(sz[i]);
    }
    int q; cin>>q;
    while (q--){
        int type,x,d,w; cin>>type;
        if (type==1){
            cin>>x>>d>>w;
            cntl[x].modify(d,1,w);
            for (int i=0;i<d;i++){
                if (x==0) break;
                int pos=loc[x];
              //  cout<<x<<" "<<fa[x]<<' '<<pos<<" "<<d-i+1<<'\n';
                cntl[fa[x]].modify(d-i-1,pos+1,w);
                cntr[fa[x]].modify(d-i-1,sz[fa[x]]-pos+1,w);
               // cout<<"dis : "<<d-i+1<<" OK\n";
                x=fa[x];
            }
        }
        else {
            cin>>x;
            int ans=h[x];
            int tans1=cntl[x].query(0,sz[x]),tans2;
            ans=(ans*tans1)%L;
         //   cout<<" -> "<<tans1<<'\n';
            for (int i=1;i<=40;i++){
                if (x==0) break;
                int pos=loc[x];
                tans1=cntl[fa[x]].query(i,pos);
                tans2=cntr[fa[x]].query(i,sz[fa[x]]-pos);
              //  cout<<" -> "<<tans1<<' '<<tans2<<'\n';
                ans=(ans*tans1%L*tans2)%L;
                x=fa[x];
            }
            cout<<ans<<'\n';
        }
    }
}

Subtask #1
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 206 ms 438648 KB Output is correct
2 Correct 199 ms 438632 KB Output is correct
3 Correct 197 ms 438692 KB Output is correct
4 Correct 227 ms 444352 KB Output is correct
5 Correct 205 ms 444608 KB Output is correct
6 Correct 200 ms 444672 KB Output is correct
7 Correct 238 ms 444856 KB Output is correct
8 Correct 205 ms 444960 KB Output is correct
9 Correct 195 ms 438704 KB Output is correct
10 Correct 204 ms 438836 KB Output is correct
11 Correct 201 ms 438732 KB Output is correct
12 Correct 198 ms 438768 KB Output is correct
13 Correct 202 ms 438808 KB Output is correct
14 Correct 198 ms 438872 KB Output is correct
15 Correct 208 ms 438732 KB Output is correct
16 Correct 222 ms 438728 KB Output is correct
17 Correct 201 ms 438788 KB Output is correct
18 Correct 221 ms 438792 KB Output is correct
19 Correct 197 ms 438676 KB Output is correct
20 Correct 208 ms 438944 KB Output is correct
21 Correct 205 ms 438780 KB Output is correct
22 Correct 197 ms 438732 KB Output is correct
23 Correct 200 ms 438692 KB Output is correct
24 Correct 197 ms 438712 KB Output is correct
25 Correct 211 ms 438732 KB Output is correct
26 Correct 219 ms 438712 KB Output is correct
27 Correct 203 ms 438860 KB Output is correct
28 Correct 198 ms 438808 KB Output is correct

Subtask #2
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 199 ms 438816 KB Output is correct
2 Runtime error 997 ms 1048576 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #3
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 199 ms 438816 KB Output is correct
2 Runtime error 997 ms 1048576 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #4
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 214 ms 438700 KB Output is correct
2 Runtime error 834 ms 1048576 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 223 ms 438676 KB Output is correct
2 Runtime error 815 ms 1048576 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 206 ms 438648 KB Output is correct
2 Correct 199 ms 438632 KB Output is correct
3 Correct 197 ms 438692 KB Output is correct
4 Correct 227 ms 444352 KB Output is correct
5 Correct 205 ms 444608 KB Output is correct
6 Correct 200 ms 444672 KB Output is correct
7 Correct 238 ms 444856 KB Output is correct
8 Correct 205 ms 444960 KB Output is correct
9 Correct 195 ms 438704 KB Output is correct
10 Correct 204 ms 438836 KB Output is correct
11 Correct 201 ms 438732 KB Output is correct
12 Correct 198 ms 438768 KB Output is correct
13 Correct 202 ms 438808 KB Output is correct
14 Correct 198 ms 438872 KB Output is correct
15 Correct 208 ms 438732 KB Output is correct
16 Correct 222 ms 438728 KB Output is correct
17 Correct 201 ms 438788 KB Output is correct
18 Correct 221 ms 438792 KB Output is correct
19 Correct 197 ms 438676 KB Output is correct
20 Correct 208 ms 438944 KB Output is correct
21 Correct 205 ms 438780 KB Output is correct
22 Correct 197 ms 438732 KB Output is correct
23 Correct 200 ms 438692 KB Output is correct
24 Correct 197 ms 438712 KB Output is correct
25 Correct 211 ms 438732 KB Output is correct
26 Correct 219 ms 438712 KB Output is correct
27 Correct 203 ms 438860 KB Output is correct
28 Correct 198 ms 438808 KB Output is correct
29 Correct 199 ms 438816 KB Output is correct
30 Runtime error 997 ms 1048576 KB Execution killed with signal 9
31 Halted 0 ms 0 KB -