結果

問題 No.235 めぐるはめぐる (5)
ユーザー hashiryohashiryo
提出日時 2018-10-10 15:33:29
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 6,368 bytes
コンパイル時間 1,850 ms
コンパイル使用メモリ 110,208 KB
実行使用メモリ 33,188 KB
最終ジャッジ日時 2023-08-02 19:18:51
合計ジャッジ時間 13,970 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
権限があれば一括ダウンロードができます

ソースコード

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 ull INF = ULLONG_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)];
  }
private:
  int t;
  void dfs_size(int v=0){
    size[v]=1;
    for(int& u:g[v]){
      if(par[u]>=0)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[2*112345];

ll mod_pow(ll x,int n){
  if(n==0)return 1;
  ll ret=mod_pow(x*x%MOD,n/2);
  if(n&1)ret=ret*x%MOD;
  return ret;
}

inline int get_min2pow(int n) {
	int res = 1;
	while (res < n)
		res *= 2;
	return res;
}
//遅延評価SegmentTree///////////
//lazyの初期値に注意
template<typename T, T dval, T dlazy>
struct delay_segtree {
	int N;
	vector<T> node, lazy;
	//例外値　ex)INF,0
	T default_value = dval;
	T default_lazy = dlazy;

	static inline T merge(const T& l, const T& r) {
		//RMQ
		//return min(l,r);
		//RSQ
		return (l + r)%MOD;
	}
	static inline void lazy_transmit(T& l, const T& r) {
		//RAQ
		l = (l+r)%MOD;
		//RUQ
		//l = r;
	}
	delay_segtree(int n) {
		N = get_min2pow(n);
		node.resize(2 * N, default_value);
		lazy.resize(2 * N, default_lazy);
	}
	delay_segtree(vector<int> v) {
		int sz = v.size();
		N = get_min2pow(sz);
		node.resize(2 * N, default_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]);
	}
	// k 番目のノードについて遅延評価を行う
	void eval(int k, int l, int r) {
		// 遅延配列が空でない場合、自ノード及び子ノードへの
		// 値の伝播が起こる
		if (lazy[k] != default_lazy) {
			lazy_transmit(node[k], lazy[k]);
			// 最下段かどうかのチェックをしよう
			//(RAQのとき
			// 子ノードは親ノードの 1/2 の範囲であるため、
			// 伝播させるときは半分にする)
			if (r - l > 1) {
				lazy_transmit(lazy[2 * k + 1], lazy[k]*(CC[(l+r)/2]-CC[l])%MOD*mod_pow(CC[r]-CC[l],MOD-2)%MOD);
				lazy_transmit(lazy[2 * k + 2], lazy[k]*(CC[r]-CC[(l+r)/2])%MOD*mod_pow(CC[r]-CC[l],MOD-2)%MOD);
			}
			// 伝播が終わったので、自ノードの遅延配列を空にする
			lazy[k] = default_lazy;
		}
	}
	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], (CC[r]-CC[l]) * x%MOD);
			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]);
		}
	}
	// add k th element
	void add(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 find(int k) {
		//遅延評価をまずしておく
		query(k, k + 1);
		return node[k + N - 1];
	}
};





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