結果

問題 No.235 めぐるはめぐる (5)
ユーザー smiken_61smiken_61
提出日時 2018-12-30 01:05:54
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,505 ms / 10,000 ms
コード長 16,493 bytes
コンパイル時間 1,994 ms
コンパイル使用メモリ 177,644 KB
実行使用メモリ 80,536 KB
最終ジャッジ日時 2024-10-04 03:35:00
合計ジャッジ時間 7,844 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,505 ms
80,536 KB
testcase_01 AC 976 ms
79,552 KB
testcase_02 AC 1,379 ms
79,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

     #include <bits/stdc++.h>

//    #include <boost/multiprecision/cpp_int.hpp>
 #define int long long
 #define inf  1000000007
 #define pa pair<int,int>
 #define ll long long
 #define pal pair<double,double>
 #define ppap pair<pa,int>
  #define PI 3.14159265358979323846
  #define paa pair<int,char>
  #define  mp make_pair
  #define  pb push_back
  #define EPS (1e-10)
                                          
    int dx[8]={0,1,0,-1,1,1,-1,-1};
    int dy[8]={1,0,-1,0,-1,1,1,-1};
                                            using namespace std;
                                   			class pa3{
                                            	public:
                                            	int x;
                                   				int y,z;
                                            	pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
                                            	bool operator < (const pa3 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		 return z<p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                   				bool operator > (const pa3 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		 return z>p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa3 &p) const{
                                            		return x==p.x && y==p.y && z==p.z;
                                            	}
                                            		bool operator != (const pa3 &p) const{
                                            			return !( x==p.x && y==p.y && z==p.z);
                                            	}
                                            
                                            };
                                            
                                            class pa4{
                                            	public:
                                            	int x;
                                            	int y,z,w;
                                            	pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
                                            	bool operator < (const pa4 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		if(z!=p.z)return z<p.z;
                                            		return w<p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator > (const pa4 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		if(z!=p.z)return z>p.z;
                                            		return w>p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa4 &p) const{
                                            		return x==p.x && y==p.y && z==p.z &&w==p.w;
                                            	}
                                            		
                                            
                                            };
                                            class pa2{
                                            	public:
                                            	int x,y;
                                            	pa2(int x=0,int y=0):x(x),y(y) {}
                                            	pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
                                            	pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
                                            	bool operator < (const pa2 &p) const{
                                            		return y != p.y ? y<p.y: x<p.x;
                                            	}
                                            	bool operator > (const pa2 &p) const{
                                            		return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa2 &p) const{
                                            		return abs(x-p.x)==0 && abs(y-p.y)==0;
                                            	}
                                            	bool operator != (const pa2 &p) const{
                                            		return !(abs(x-p.x)==0 && abs(y-p.y)==0);
                                            	}
                                            		
                                            
                                            };
                                            
/*
                                            class Point{
                                            	public:
                                            	double x,y;
                                            	Point(double x=0,double y=0):x(x),y(y) {}
                                            	Point operator + (Point p) {return Point(x+p.x,y+p.y);}
                                            	Point operator - (Point p) {return Point(x-p.x,y-p.y);}
                                            	Point operator * (double a) {return Point(x*a,y*a);}
                                            	Point operator / (double a) {return Point(x/a,y/a);}
                                            	double absv() {return sqrt(norm());}
                                            	double norm() {return x*x+y*y;}
                                            	bool operator < (const Point &p) const{
                                            		return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const Point &p) const{
                                            		return fabs(x-p.x)<EPS && fabs(y-p.y)<EPS;
                                            	}
                                            };
                                            typedef Point Vector;
                                     #define pl pair<int,pas>
                                            struct Segment{
                                            Point p1,p2;
                                            };
                                             double dot(Vector a,Vector b){
                                            	return a.x*b.x+a.y*b.y;
                                            }
                                            double cross(Vector a,Vector b){
                                            	return a.x*b.y-a.y*b.x;
                                            }
                                        
                bool parareru(Point a,Point b,Point c,Point d){
                //	if(abs(cross(a-b,d-c))<EPS)cout<<"dd "<<cross(a-b,d-c)<<endl;
                	return abs(cross(a-b,d-c))<EPS;
                }
                double distance_ls_p(Point a, Point b, Point c) {
                  if ( dot(b-a, c-a) < EPS ) return (c-a).absv();
                  if ( dot(a-b, c-b) < EPS ) return (c-b).absv();
                  return abs(cross(b-a, c-a)) / (b-a).absv();
                }
                bool is_intersected_ls(Segment a,Segment b) {
                	if(a.p1==b.p1||a.p2==b.p1||a.p1==b.p2||a.p2==b.p2) return false;
                	if(parareru((a.p2),(a.p1),(a.p1),(b.p2))&&parareru((a.p2),(a.p1),(a.p1),(b.p1))){
                //		cout<<"sss"<<endl;
                		if(dot(a.p1-b.p1,a.p1-b.p2)<EPS) return true;
                		if(dot(a.p2-b.p1,a.p2-b.p2)<EPS) return true;
                		if(dot(a.p1-b.p1,a.p2-b.p1)<EPS) return true;
                		if(dot(a.p1-b.p2,a.p2-b.p2)<EPS) return true;
                		return false;
                	}
                  else return ( cross(a.p2-a.p1, b.p1-a.p1) * cross(a.p2-a.p1, b.p2-a.p1) < EPS ) && ( cross(b.p2-b.p1, a.p1-b.p1) * cross(b.p2-b.p1, a.p2-b.p1) < EPS );
                }
                 
                double segment_dis(Segment a,Segment b){
                	if(is_intersected_ls(a,b))return 0;
                	double r=distance_ls_p(a.p1, a.p2, b.p1);
                	r=min(r,distance_ls_p(a.p1, a.p2, b.p2));
                	r=min(r,distance_ls_p(b.p1, b.p2, a.p2));
                	r=min(r,distance_ls_p(b.p1, b.p2, a.p1));
                	return r;
                }
                Point intersection_ls(Segment a, Segment b) {
                  Point ba = b.p2-b.p1;
                  double d1 = abs(cross(ba, a.p1-b.p1));
                  double d2 = abs(cross(ba, a.p2-b.p1));
                  double t = d1 / (d1 + d2);
                 
                  return a.p1 + (a.p2-a.p1) * t;
                }
 */                
                                string itos( int i ) {
                                ostringstream s ;
                                s << i ;
                                return s.str() ;
                                }
                                 
                                int gcd(int v,int b){
                                	if(v>b) return gcd(b,v);
                                	if(v==b) return b;
                                	if(b%v==0) return v;
                                	return gcd(v,b%v);
                                }
                 
                                double distans(double x1,double y1,double x2,double y2){
                                	double rr=(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
                                	return sqrt(rr);
                                	
                                }
                                int mod;
int extgcd(int a, int b, int &x, int &y) {
    if (b == 0) {
        x = 1;
        y = 0;
        return a;
    }
    int d = extgcd(b, a%b, y, x);
    y -= a/b * x;
    return d;
}
pa operator+(const pa & l,const pa & r) {   
    return {l.first+r.first,l.second+r.second};                                    
}    
                                
                int pr[1000010];
                int inv[1000010];
                
                int beki(int wa,int rr,int warukazu){
                	if(rr==0) return 1%warukazu;
                	if(rr==1) return wa%warukazu;
                	wa%=warukazu;
                	if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
                	ll zx=beki(wa,rr/2,warukazu);
                	return (zx*zx)%warukazu;
                }
    double bekid(double w,int r){
    	if(r==0) return 1.0;
    	if(r==1) return w;
    	if(r%2) return bekid(w,r-1)*w;
    	double f=bekid(w,r/2);
    	return f*f;
    }
                
    			int comb(int nn,int rr){
    				int r=pr[nn]*inv[rr];
    				r%=mod;
    				r*=inv[nn-rr];
    				r%=mod;
    				return r;
    			}
                
                void gya(int ert){
                	pr[0]=1;
                	for(int i=1;i<ert;i++){
                		pr[i]=(pr[i-1]*i)%mod;
                	}
                	for(int i=0;i<ert;i++) inv[i]=beki(pr[i],mod-2,mod);
                	
                }
                
              //   cin.tie(0);
    		//	ios::sync_with_stdio(false);
    			//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;            
                 //sort(ve.begin(),ve.end(),greater<int>());
    
                                     //----------------kokomade tenpure------------

vector<int> G[200020],G2[200020];
int cont=0;
int sz[200020];
int in[200020];
int rin[200020];
int nxt[200020];
int out[200020];
void dfs_sz(int v){
    sz[v]=1;
    for(auto &u :G2[v]){
        dfs_sz(u);
        sz[v]+=sz[u];
        if(sz[u]>sz[G2[v][0]])swap(u,G2[v][0]);
    }
}
void dfs_hld(int v){
    if(v==1) nxt[v]=1;
    in[v]=cont;
    cont++;
    rin[in[v]]=v;
    for(auto u:G2[v]){
        if(u==G2[v][0])nxt[u]=nxt[v];
        else nxt[u]=u;
        dfs_hld(u);
    }
    out[v]=cont;
}
// vのsubtreeに含まれる区間は[in[v],out[v])
// vを含むheavy列の一番上はnxt[v]
// heavy列で[in[nxt[v]],...,in[v]]に一様操作すればOK
 
 
struct seg_addsum_getsum{
	//       1
	//   2        3
	// 4   5   6    7
	
	
	private:
	public:
	
	// (1<<15)=32768
	// 1<<16 = 65536
	// 1<<17 = 131072
	// 1<<18 = 262144
	
	int cor=(1<<18);
	
	vector<int> vec;
	vector<int> lazy;
	vector<int> fre;
	void shoki1(){
		vec.resize(2*cor+3, 0);
		lazy.resize(2*cor+3, 0);
		fre.resize(2*cor+3, 0);
	
	}
	void shoki2(){
			for(int i=cor-1;i>0;i--) vec[i]=(vec[2*i]+vec[2*i+1])%inf;
			for(int i=cor-1;i>0;i--) fre[i]=(fre[2*i]+fre[2*i+1])%inf;
	}
		
	void chien(int k,int l,int r){
		lazy[k]%=inf;
		if(lazy[k]==0) return;
		if(l+1<r){
			lazy[k*2]+=lazy[k];
			lazy[k*2]%=inf;
			lazy[k*2+1]+=lazy[k];
			lazy[k*2+1]%=inf;
		}
		vec[k]+=(lazy[k]*fre[k])%inf;
		vec[k]%=inf;
		lazy[k]=0;
	}
	
	void rangeadd(int a,int b,int w, int k=1,int l=0,int r=-3){
		//[a,b)に w加算
		if(r<0) r=cor;
		chien(k,l,r);
		if(a<=l && r<=b){
			lazy[k]+=w;
			lazy[k]%=inf;
			chien(k,l,r);
			return ;
		}
		if(r<=a || b<=l){
			return ;
		}
		
		rangeadd(a,b,w,k*2,l,(l+r)/2);
		rangeadd(a,b,w,k*2+1,(l+r)/2,r);
	vec[k]=vec[2*k]+vec[2*k+1];
	vec[k]%=inf;
		return ;
	}
	
	// [a,b)
	// k-th node
	// k no kukanha [l,r)
	int getsum(int a,int b,int k=1,int l=0,int  r=-3){
		if(r<0) r=cor;
		chien(k,l,r);
		if(a<=l && r<=b){
			return vec[k];
		}
		if(r<=a || b<=l){
			return 0;
		}
		return (getsum(a,b,k*2,l,(l+r)/2)+getsum(a,b,k*2+1,(l+r)/2,r))%inf;
		
	}
	
	void pre(){
		for(int i=1;i<2*cor;i++){
			cout<<vec[i]<<"/"<<lazy[i]<<" ";
			if(((i+1)&(-(i+1)))==i+1) cout<<endl;
		}
	}
	
};
seg_addsum_getsum SE;

int depth[200020];
int s[200200],c[200020];
int dp[200020][20];
void fdsf(int r,int p,int d){
	depth[r]=d;
	dp[r][0]=p;
	for(auto v:G[r])if(v!=p){
		G2[r].pb(v);
		fdsf(v,r,d+1);
	}
}

pa4  lca(int a,int b){
	int aa=a,bb=b;
	bool sw=0;
	if(depth[b]>depth[a]) swap(a,b),sw=1;
	int sa=depth[a]-depth[b];
	for(int i=19;i>=0;i--)if(sa&(1<<i))a=dp[a][i];
	if(a==b){
		if(sw==0){
			return {aa,a,-1,-1};
		}
		else{
			return {bb,a,-1,-1};
		}
	}
	for(int i=19;i>=0;i--){
		if(dp[a][i]!=dp[b][i]){
			a=dp[a][i];
			b=dp[b][i];
			
		}
	}
	if(sw==0){
		return {aa,dp[a][0],bb,b};
	}
	return {bb,dp[a][0],aa,b};
//	return dp[a][0];
}

void tasu(int x,int y,int z){
	int sita=x;
	while(1){
		int ue=nxt[sita];
		if(ue==nxt[y]){
			SE.rangeadd(in[y],in[sita]+1,z);
			return;
		}
		SE.rangeadd(in[ue],in[sita]+1,z);
		sita=dp[ue][0];
	}
}

int solve(int r){
//	cout<<"sol"<<r<<endl;
	int sita=r;
	int ans=0;
	while(1){
//		cout<<sita<<" "<<endl;
		int ue=nxt[sita];
		ans+=SE.getsum(in[ue],in[sita]+1);
	//	cout<<"ue "<<ue<<" "<<sita<<" "<<SE.getsum(in[ue],in[sita]+1)<<endl;
		ans%=inf;
		if(ue==1) return ans;
		sita=dp[ue][0];
		
	}
}
 signed main(){

    	       cin.tie(0);
   			ios::sync_with_stdio(false);
 	int n;
 	cin>>n;
 	for(int i=1;i<=n;i++)cin>>s[i];

 	for(int i=1;i<=n;i++)cin>>c[i];
 	
 	for(int i=0;i<n-1;i++){
 		int y,yy;
 		cin>>y>>yy;
 		G[y].pb(yy);
 		G[yy].pb(y);
 	}
 	
 	fdsf(1,1,0);
 	
 	for(int i=1;i<20;i++)for(int j=1;j<=n;j++){
 		dp[j][i]=dp[dp[j][i-1]][i-1];
 	}
 	
 	dfs_sz(1);
 	dfs_hld(1);
 //	for(int i=1;i<=n;i++)cout<<nxt[i]<<" ";
 //	cout<<endl;
 	SE.shoki1();
 	for(int i=1;i<=n;i++){
 		SE.vec[SE.cor+in[i]]=s[i];
 		
 		SE.fre[SE.cor+in[i]]=c[i];
 	}
 	SE.shoki2();
 	
 	int q;
 	cin>>q;
 
 	for(int er=0;er<q;er++){
 		int t;
 //		cout<<"que "<<er<<endl;
 		cin>>t;
 		if(t){
 			int x,y;
 			cin>>x>>y;
 			int l=lca(x,y).y;
 //			cout<<x<<" "<<y<<" "<<l<<endl;
 			int ans=0;
 			ans+=solve(x);
 			
 			ans+=solve(y);
 //			cout<<y<<" "<<solve(y)<<endl;
 			ans+=inf-solve(l);
 			
 			if(l!=1){
 				ans+=inf-solve(dp[l][0]);
 			}
 			cout<<ans%inf<<endl;
 		}
 		else{
 			int x,y,z;
 			cin>>x>>y>>z;
 			pa4 e=lca(x,y);
 			tasu(e.x,e.y,z);
 			if(e.w>=0) tasu(e.z,e.w,z);
 			
 		}
 	}
 	return 0;
  }

0