結果

問題 No.235 めぐるはめぐる (5)
ユーザー hashiryohashiryo
提出日時 2018-10-13 18:01:22
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,527 ms / 10,000 ms
コード長 6,716 bytes
コンパイル時間 1,038 ms
コンパイル使用メモリ 99,948 KB
実行使用メモリ 33,388 KB
最終ジャッジ日時 2024-04-20 20:09:04
合計ジャッジ時間 7,948 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,527 ms
32,552 KB
testcase_01 AC 1,121 ms
33,388 KB
testcase_02 AC 1,463 ms
32,868 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <stdio.h>
#include <fstream>
#include <algorithm>
#include <string>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <vector>
#include <limits.h>
#include <math.h>
#include <functional>
#include <bitset>

#define repeat(i,n) for (long long i = 0; (i) < (n); ++ (i))
#define debug(x) cerr << #x << ": " << x << '\n'
#define debugArray(x,n) for(long long i = 0; (i) < (n); ++ (i)) cerr << #x << "[" << i << "]: " << x[i] << '\n'
#define debugArrayP(x,n) for(long long i = 0; (i) < (n); ++ (i)) cerr << #x << "[" << i << "]: " << x[i].first<< " " << x[i].second << '\n'

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> Pii;
typedef vector<int> vint;
typedef vector<ll> vll;
const ll INF = INT_MAX;
const ll MOD = 1e9+7;

struct HLDecomposition{
public:
    int V;
    vector<vint> g;
    vint dep,par,head,size,inv;
    vint in,out;
    HLDecomposition(int size_)
    :V(size_),g(V),dep(V,0),par(V,-1),head(V),size(V),inv(V),in(V),out(V),t(0){}

    void add_edge(int u,int v){
        g[u].push_back(v);
        g[v].push_back(u);
    }
    void build(int root=0){
        par[root]=0;
        dfs_size(root);
        par[root]=-1;
        dfs_hld(root);
    }

    int lca(int u,int v){
        while(1){
            if(in[u]>in[v])swap(u,v);
            if(head[u]==head[v])return u;
            v=par[head[v]];
        }
    }

    int distance(int u,int v){
        return dep[u]+dep[v]-2*dep[lca(u,v)];
    }

    int operator[](const int &k){
        return in[k];
    }
private:
    int t;
    void dfs_size(int v=0){
        size[v]=1;
        for(int& u:g[v]){
            if(par[u]>=0){
                swap(u,g[v].back());
                if(u==g[v].back())continue;
            }
            par[u]=v;
            dfs_size(u);
            size[v]+=size[u];
            if(size[u]>size[g[v][0]]){
                swap(u,g[v][0]);
            }
        }
    }
    void dfs_hld(int v=0){
        in[v] = t++;
        inv[in[v]]=v;
        for(int& u:g[v]){
            if(par[u]!=v)continue;
            head[u]=(u==g[v][0]?head[v]:u);
            dfs_hld(u);
        }
        out[v]=t;
    }

};

ll CC[212345];

//遅延評価SegmentTree///////////
//lazyの初期値に注意
template<typename T>
struct delay_segtree {
private:
    int N;
    vector<T> node, lazy;
    //-------書くとこ---------------------
    //例外値　ex)INF,0
    T init_value = 0;//初期値
    T default_value = 0;//単位元
    T default_lazy = 0;
    static inline T merge(const T& u, const T& v) {
        return (u+v)%MOD;
    }
    static inline void lazy_transmit(T& u, const T& v) {
        u = (u+v)%MOD;
    }
    static inline void node_transmit(T& u,const T& v,int l,int r){
        u = (u+(CC[r]-CC[l]+MOD)*v%MOD)%MOD;
    }
    //------------------------------------
    // k 番目のノードについて遅延評価を行う
    void eval(int k, int l, int r) {
        // 遅延配列が空でない場合、自ノード及び子ノードへの
        // 値の伝播が起こる
        if (lazy[k] != default_lazy) {
            node_transmit(node[k], lazy[k],l,r);
            // 最下段かどうかのチェックをしよう
            if (r - l > 1) {
                lazy_transmit(lazy[2 * k + 1], lazy[k]);
                lazy_transmit(lazy[2 * k + 2], lazy[k]);
            }
            // 伝播が終わったので、自ノードの遅延配列を空にする
            lazy[k] = default_lazy;
        }
    }
public:
    delay_segtree(int n) {
        for(N=1;N<n;N*=2);
        node.resize(2 * N, init_value);
        lazy.resize(2 * N, default_lazy);
    }
    delay_segtree(vector<int> v) {
        int sz = v.size();
        for(N=1;N<sz;N*=2);
        node.resize(2 * N, init_value);
        lazy.resize(2 * N, default_lazy);
        for (int i = 0; i < sz; i++)
            node[i + N - 1] = v[i];
        for (int i = N - 2; i >= 0; i--)
            node[i] = merge(node[2 * i + 1], node[2 * i + 2]);
    }
    void update(int a, int b, T x, int k = 0, int l = 0, int r = -1) {
        if (r < 0)
            r = N;
        // k 番目のノードに対して遅延評価を行う
        eval(k, l, r);
        // 範囲外なら何もしない
        if (b <= l || r <= a)
            return;
        // 完全に被覆しているならば、遅延配列に値を入れた後に評価
        if (a <= l && r <= b) {
            lazy_transmit(lazy[k], x);
            eval(k, l, r);
        }
        // そうでないならば、子ノードの値を再帰的に計算して、
        // 計算済みの値をもらってくる
        else {
            update(a, b, x, 2 * k + 1, l, (l + r) / 2);
            update(a, b, x, 2 * k + 2, (l + r) / 2, r);
            node[k] = merge(node[2 * k + 1], node[2 * k + 2]);
        }
    }
    // update k th element
    void update(int k, T val) {
        k += N - 1; // leaf
        node[k] = val;
        while (k > 0) {
            k = (k - 1) / 2;
            node[k] = merge(node[k * 2 + 1], node[k * 2 + 2]);
        }
    }
    // [a, b)
    T query(int a, int b, int k = 0, int l = 0, int r = -1) {
        if (r < 0)
            r = N;
        eval(k, l, r);
        if (r <= a or b <= l)
            return default_value;
        if (a <= l and r <= b)
            return node[k];
        int m = (l + r) / 2;
        T vl = query(a, b, k * 2 + 1, l, m);
        T vr = query(a, b, k * 2 + 2, m, r);
        return merge(vl, vr);
    }

    T operator[](const int &k) {
        //遅延評価をまずしておく
        query(k, k + 1);
        return node[k + N - 1];
    }
};




int main(){
  int N;cin>>N;
  vll S(N),C(N);
  repeat(i,N){
    cin>>S[i];
  }
  repeat(i,N){
    cin>>C[i];
  }
  HLDecomposition hld(N);
  repeat(i,N-1){
    int A,B;cin>>A>>B;A--;B--;
    hld.add_edge(A,B);
  }
  hld.build();
  delay_segtree<ll> seg(N);
  repeat(i,N){
    seg.update(hld[i],S[i]);
    CC[hld[i]+1]=C[i];
  }
  repeat(i,N){
    CC[i+1] = (CC[i+1]+CC[i])%MOD;
  }
  //debugArray(CC,N);
  int Q;cin>>Q;
  repeat(q,Q){
    int op,X,Y;cin>>op>>X>>Y;X--;Y--;
    if(op){
      ll ans = 0;
      while(1){
        if(hld[X]>hld[Y])swap(X,Y);
        ans = (ans+seg.query(max(hld[X],hld[hld.head[Y]]),hld[Y]+1))%MOD;
        if(hld.head[X]!=hld.head[Y])Y=hld.par[hld.head[Y]];
        else break;
      }
      cout << ans <<endl;
    }else{
      int Z;cin>>Z;
      while(1){
        if(hld[X]>hld[Y])swap(X,Y);
        seg.update(max(hld[X],hld[hld.head[Y]]),hld[Y]+1,Z);
        if(hld.head[X]!=hld.head[Y])Y=hld.par[hld.head[Y]];
        else break;
      }
      //debugArray(seg,N);
    }
  }
  return 0;
}
0