結果
問題 | No.235 めぐるはめぐる (5) |
ユーザー | smiken_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 |
ソースコード
#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))&¶reru((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; }