Submission #785639

#	Time	Username	Problem	Language	Result	Execution time	Memory
785639		devariaota	Sprinkler (JOI22_sprinkler)	C++17	0 / 100	4058 ms	17420 KiB

sprinkler

#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define keish                             ios_base::sync_with_stdio(0);       cin.tie(0); cout.tie(0)
      
using namespace std;

int n, l, q, u, v, t, x, d, w, lim;

signed main(){
      keish;
      cin >> n >> l;
      vector<vector<int>> e(n);
      for(int i = 1; i < n; i++){
            cin >> u >> v; u--, v--;
            e[u].push_back(v);
            e[v].push_back(u);
      }

      vector<int> a(n);
      for(auto &x : a) cin >> x;

      function<void(int, int)> dfs = [&](int u, int d){
            if(d > lim) return;
            for(auto v : e[u]){
                  (a[v] *= w) %= l;
                  dfs(v, d + 1);
            }
      };

      cin >> q;
      while(q--){
            cin >> t;
            if(t == 1){
                  cin >> x >> lim >> w; x--;
                  (a[x] *= w) %= l;
                  if(lim > 0){
                        dfs(x, 1);
                  }
            }else{      
                  cin >> x; x--;
                  cout << a[x] << '\n';
            }
      }
}     

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