oj.uz

Submission #637963

# Submission time Handle Problem Language Result Execution time Memory
637963 2022-09-03T18:27:01 Z victor_gao Sprinkler (JOI22_sprinkler) C++17
3 / 100
 4000 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[44];
    int n=0,m=43;
    void init(int _n){
        n=_n;
        for (int i=0;i<=m;i++)
            bit[i].resize(n+1,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;
    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 190 ms 429320 KB Output is correct
2 Correct 189 ms 429308 KB Output is correct
3 Correct 195 ms 429364 KB Output is correct
4 Correct 216 ms 432220 KB Output is correct
5 Correct 197 ms 432544 KB Output is correct
6 Correct 196 ms 432716 KB Output is correct
7 Correct 201 ms 432716 KB Output is correct
8 Correct 195 ms 432856 KB Output is correct
9 Correct 188 ms 429244 KB Output is correct
10 Correct 192 ms 429244 KB Output is correct
11 Correct 189 ms 429336 KB Output is correct
12 Correct 195 ms 429352 KB Output is correct
13 Correct 192 ms 429288 KB Output is correct
14 Correct 190 ms 429232 KB Output is correct
15 Correct 189 ms 429260 KB Output is correct
16 Correct 194 ms 429324 KB Output is correct
17 Correct 194 ms 429456 KB Output is correct
18 Correct 189 ms 429320 KB Output is correct
19 Correct 188 ms 429260 KB Output is correct
20 Correct 188 ms 429292 KB Output is correct
21 Correct 196 ms 429340 KB Output is correct
22 Correct 190 ms 429264 KB Output is correct
23 Correct 192 ms 429324 KB Output is correct
24 Correct 192 ms 429384 KB Output is correct
25 Correct 190 ms 429340 KB Output is correct
26 Correct 191 ms 429372 KB Output is correct
27 Correct 193 ms 429340 KB Output is correct
28 Correct 189 ms 429284 KB Output is correct

Subtask #2
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 211 ms 429284 KB Output is correct
2 Runtime error 986 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 211 ms 429284 KB Output is correct
2 Runtime error 986 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 192 ms 429260 KB Output is correct
2 Execution timed out 4075 ms 1017156 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 198 ms 429384 KB Output is correct
2 Execution timed out 4101 ms 1011348 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 190 ms 429320 KB Output is correct
2 Correct 189 ms 429308 KB Output is correct
3 Correct 195 ms 429364 KB Output is correct
4 Correct 216 ms 432220 KB Output is correct
5 Correct 197 ms 432544 KB Output is correct
6 Correct 196 ms 432716 KB Output is correct
7 Correct 201 ms 432716 KB Output is correct
8 Correct 195 ms 432856 KB Output is correct
9 Correct 188 ms 429244 KB Output is correct
10 Correct 192 ms 429244 KB Output is correct
11 Correct 189 ms 429336 KB Output is correct
12 Correct 195 ms 429352 KB Output is correct
13 Correct 192 ms 429288 KB Output is correct
14 Correct 190 ms 429232 KB Output is correct
15 Correct 189 ms 429260 KB Output is correct
16 Correct 194 ms 429324 KB Output is correct
17 Correct 194 ms 429456 KB Output is correct
18 Correct 189 ms 429320 KB Output is correct
19 Correct 188 ms 429260 KB Output is correct
20 Correct 188 ms 429292 KB Output is correct
21 Correct 196 ms 429340 KB Output is correct
22 Correct 190 ms 429264 KB Output is correct
23 Correct 192 ms 429324 KB Output is correct
24 Correct 192 ms 429384 KB Output is correct
25 Correct 190 ms 429340 KB Output is correct
26 Correct 191 ms 429372 KB Output is correct
27 Correct 193 ms 429340 KB Output is correct
28 Correct 189 ms 429284 KB Output is correct
29 Correct 211 ms 429284 KB Output is correct
30 Runtime error 986 ms 1048576 KB Execution killed with signal 9
31 Halted 0 ms 0 KB -