Submission #1007626

#	Time	Username	Problem	Language	Result	Execution time	Memory
1007626		modwwe	Sprinkler (JOI22_sprinkler)	C++17	100 / 100	484 ms	102224 KiB

sprinkler

//https://www.instagram.com/_modwwe/
#pragma GCC optimize("Ofast,unroll-loops")
//#pragma GCC target("avx2,bmi,bmi2")
#include<bits/stdc++.h>
#define int long long
//#define ll long long
#define down cout<<'\n';
#define NHP     ios_base::sync_with_stdio(0);cout.tie(0);cin.tie(0);
#define modwwe  int t;cin>>t; while(t--)
#define bit(i,j) (i>>j&1)
#define sobit(a) __builtin_popcountll(a)
#define task "test"
#define fin(x) freopen(x".inp","r",stdin)
#define fou(x) freopen(x".out","w",stdout)
#define pb push_back
#define checktime   cerr << (double)clock() / CLOCKS_PER_SEC * 1000  << " ms";
using namespace std;
void phongbeo();
const int mod2=1e9+7;
const int  mod1=998244353;
struct icd
{
    int a,b;
};
struct ib
{
    int a;
    int b;
};
struct ic
{
    int a,b,c;
};
struct id
{
    int a,b,c,d;
};
struct ie
{
    int a,b,c, d,e,f;

};
int n,m,s1,s2,s4,s3,sf,k,r,mid,s5,s6,mx,s7,s8,s9,mx2,res,dem2=0,dem=0,l;
int  i,s10,s12;
int el=29;
main()
{
#ifndef ONLINE_JUDGE
///     fin(task),fou(task);
#endif
    NHP;
    /// cin>>s1;
   // modwwe
    phongbeo(),down
}
vector<int> v[400001];
int cha[400001];
int val[400001][42];
void dfs(int x,int y)
{
     cha[x]=y;
      for(auto f:v[x])
        if(f!=y) dfs(f,x);
}
void dfs_upd(int x,int y,int z)
{ if(y==-1||x==0)return;
val[x][y]=(val[x][y]*z)%k;
    dfs_upd(cha[x],y-1,z);
}
void dfs_get(int x)
{
    if(dem==41) return;
if(x==1) {for(int j=dem;j<=40;j++)s2=(s2*val[x][j])%k;return;}
    else
 s2=((s2*val[x][dem])%k*val[x][dem+1])%k;
    dem++;
 dfs_get(cha[x]);
}
void phongbeo()
{
    cin>>n>>k;
for(int i=1;i<n;i++)
     cin>>l>>r,v[l].pb(r),v[r].pb(l);
dfs(1,0);
for(int i=0;i<=n;i++)
    for(int j=0;j<=41;j++)
    val[i][j]=1;
 for(int i=1;i<=n;i++)
 cin>>l,val[i][0]=l;
    cin>>m;
    for(int i=1;i<=m;i++)
    {
       cin>>l;
       if(l==1)
       {
           cin>>l>>r>>s2;
            dfs_upd(l,r,s2);
       }
       else
       { dem=0;
       s2=1;
            cin>>l;
            dfs_get(l);
   cout<<s2,down//<<" "<<dem<<" cucu",down

        }
    }
}
/*
4 7
1 2
2 3
3 4
1
1
1
1
9
2 1
2 2
2 3
2 4
1 4 10 2
2 1
2 2
2 3
2 4

*/

Compilation message (stderr)

sprinkler.cpp:46:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   46 | main()
      | ^~~~

• Subtask 1: 3 / 3
• Subtask 2: 9 / 9
• Subtask 3: 29 / 29
• Subtask 4: 12 / 12
• Subtask 5: 30 / 30
• Subtask 6: 17 / 17